pandas 自定义函数

在 pandas 中，我们可以使用 apply() 方法自定义函数来对 DataFrame 中的每个元素进行操作。具体步骤如下：

1. 定义自定义函数，该函数需要有一个输入参数，代表 DataFrame 中的每个元素；
2. 使用 apply() 方法将自定义函数应用到 DataFrame 中的每个元素；
3. 将 apply() 方法返回的结果赋值给 DataFrame 中相应的位置。

下面是一个示例代码：

import pandas as pd

# 定义自定义函数
def add_one(x):
    return x + 1

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

# 使用 apply() 方法将自定义函数应用到 DataFrame 中的每个元素
df = df.applymap(add_one)

# 输出结果
print(df)

运行结果如下：

   A  B
0  2  5
1  3  6
2  4  7

在上面的示例代码中，我们定义了一个自定义函数 add_one()，该函数将输入参数加 1。然后，我们使用 apply() 方法将该函数应用到 DataFrame 中的每个元素。最后，我们将 apply() 方法返回的结果赋值给 DataFrame 中相应的位置，得到了新的 DataFrame。