Pandas中怎么应用自定义函数

378
2024/5/14 16:55:49
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在Pandas中应用自定义函数通常使用apply方法。下面是一个简单的示例:

假设有一个包含一列数字的DataFrame:

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3, 4, 5]})

现在我们定义一个自定义函数,例如计算数字的平方:

def square(x):
    return x ** 2

然后我们可以使用apply方法将这个函数应用到DataFrame的列上:

df['B'] = df['A'].apply(square)

这样就会得到一个新的列’B’,里面存放的是’A’列中每个数字的平方值。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: pandas dataframe常用操作有哪些