首页  >  文章  >  后端开发  >  如何使用自定义函数从 Pandas DataFrame 中高效提取多列?

如何使用自定义函数从 Pandas DataFrame 中高效提取多列?

Patricia Arquette
Patricia Arquette原创
2024-11-03 00:55:29594浏览

How to Efficiently Extract Multiple Columns from a Pandas DataFrame using a Custom Function?

使用 Pandas 函数提取多列

本问题探讨了使用自定义函数从 pandas DataFrame 中提取多列的问题。函数的返回类型会出现问题,因为它需要与所需的输出正确对齐。

最初,推荐的方法是使用 df.iterrows() 迭代行。然而,后来发现这种方法明显慢得多。因此,作者选择将函数拆分为六个不同的 map(lambda ...) 调用来提取所需的列。

更有效的方法是利用 zip 函数来分配自定义函数的输出同时到多个列。使用一个示例来说明此方法,其中将名为 powers 的函数应用于一列数字。该函数计算每个数字的六个幂值,并将结果分配给 DataFrame 中的六个新列。

这种方法既优雅又高效,并且避免了迭代 DataFrame 的行。这是基于自定义函数从 DataFrame 中提取多列的推荐技术。

以上是如何使用自定义函数从 Pandas DataFrame 中高效提取多列?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn