给DataFrame的apply调用的函数添加多个参数

最新推荐文章于 2025-01-03 20:45:00 发布

原创最新推荐文章于 2025-01-03 20:45:00 发布 · 2w 阅读

24 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

python

当前文章被收录于：

python

55 篇文章 2 人学习

订阅专栏查看详情

技巧

当前文章被收录于：

技巧

34 篇文章 2 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍如何在Pandas中使用apply函数处理DataFrame，特别是当需要传递额外参数到自定义函数时的方法。通过实例演示了如何正确设置args参数以实现函数的灵活调用。

apply在调用函数时，默认将DataFrame逐行或逐列作为一个参数代入调用的函数中，但有的函数需要多个参数，这时可以在apply里加上args=，将其他参数组成一个元组代入函数中。
但有个前提：被调用的函数第一个参数必须是DataFrame的行或列，args=的元组中参数位置必须和函数定义的参数位置一致

df = pd.DataFrame([[1, 2, 3, 4], [5, 6, 7, 8]],
                  index=list('AB'),
                  columns=list('abcd'))
print(df)
'''
	a	b	c	d
A	1	2	3	4
B	5	6	7	8
'''

def test(x, y, z):
    return x + y + z
print(df.apply(test, args=(10, 100)))
'''
     a    b    c    d
A  111  112  113  114
B  115  116  117  118
'''