首页 > 后端开发 > Python教程 > 如何将自定义函数应用于多个 Pandas DataFrame 列以创建新列?

如何将自定义函数应用于多个 Pandas DataFrame 列以创建新列?

Linda Hamilton
发布: 2024-12-27 12:16:14
原创
788 人浏览过

How Can I Apply a Custom Function to Multiple Pandas DataFrame Columns to Create a New Column?

将函数应用于 Pandas DataFrame 中的多个列

考虑以下情况,其中您有一个包含多个列和自定义函数的 DataFrame您想要按元素应用于这些列以创建新的

问题:

假设您有一个包含三列的 DataFrame:

df = pd.DataFrame({'ID':['1','2','3'], 'col_1': [0,2,3], 'col_2':[1,4,5]})
登录后复制

和一个函数:

def get_sublist(sta, end):
    return mylist[sta:end+1]
登录后复制

您的目标是将 get_sublist 应用于“col_1”和“col_2” columns of df 以获得包含 'mylist' 相应子列表的新列 'col_3'。

解决方案:

要实现此目的,您可以使用以下命令Pandas 操作:

df['col_3'] = df.apply(lambda x: get_sublist(x.col_1, x.col_2), axis=1)
登录后复制

以下是code:

  • df.apply():将给定函数按元素应用到 DataFrame 的每一行。
  • lambda x:接受单个参数 x 的匿名函数,它代表 DataFrame 的一行。
  • x:包含“col_1”和“col_2”列值的 DataFrame 行对象当前迭代。
  • get_sublist(x.col_1, x.col_2):使用当前行中的“col_1”和“col_2”值调用 get_sublist 函数。
  • axis=1:指定该函数应应用于 DataFrame 的每一行,从而产生一个新的

使用此方法,生成的 DataFrame 将具有以下结构:

  ID  col_1  col_2            col_3
0  1      0      1       ['a', 'b']
1  2      2      4  ['c', 'd', 'e']
2  3      3      5  ['d', 'e', 'f']
登录后复制

请注意,此解决方案允许您将任何自定义函数传递给 apply() 方法并将其同时应用于多个列。通过 x.col_name 访问列值,您可以安全地使用列名称,即使它们包含空格或与现有 DataFrame 属性同名。

以上是如何将自定义函数应用于多个 Pandas DataFrame 列以创建新列?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板