pandas自定义函数

在 Pandas 中，我们可以使用 apply() 函数来应用自定义函数，apply() 可以对 DataFrame 和 Series 中的元素进行批量操作。自定义函数可以是一个 lambda 函数或一个普通的 Python 函数。

假设有一个 DataFrame 如下：

import pandas as pd

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

print(df)

输出结果为：

   A  B
0  1  4
1  2  5
2  3  6

我们可以使用 apply() 函数对 DataFrame 中的每个元素进行平方操作，示例代码如下：

import pandas as pd

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

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

# 应用自定义函数
df = df.apply(square)

print(df)

输出结果为：

   A   B
0  1  16
1  4  25
2  9  36

也可以使用 lambda 函数来应用自定义函数：

import pandas as pd

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

# 使用 lambda 函数
df = df.apply(lambda x: x ** 2)

print(df)

输出结果为：

   A   B
0  1  16
1  4  25
2  9  36