温馨提示×

Pandas中怎么应用自定义函数

小亿
108
2024-05-11 16:55:49
栏目: 编程语言

在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’列中每个数字的平方值。

0