dataframe apply自定义函数

DataFrame的apply方法可以用于对DataFrame的行或列应用自定义函数。你可以通过指定`axis`参数来选择是按行还是按列应用函数。当`axis=0`时，函数将按列应用；当`axis=1`时，函数将按行应用。

下面是一个例子，演示如何使用apply方法来对DataFrame的一列应用自定义函数：

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

# 定义一个自定义函数
def square(x):
    return x ** 2

# 对DataFrame的一列应用自定义函数
df['A_squared'] = df['A'].apply(square)

print(df)

输出结果为：

   A  B  A_squared
0  1  4          1
1  2  5          4
2  3  6          9

在上面的例子中，我们定义了一个自定义函数`square`，它接受一个参数并返回参数的平方。然后，我们使用`apply`方法将这个自定义函数应用到了DataFrame的列`A`上，并将结果保存到了新的列`A_squared`中。