举例说明apply函数的用法

下面是一个简单的例子，展示了如何使用 Pandas 库中的 `apply` 函数。

假设有一个名为 `df` 的 DataFrame，其中包含两列数据 `A` 和 `B`，我们需要将这两列数据相加得到新的一列 `C`。我们可以使用 `apply` 函数来实现：

import pandas as pd

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

# 定义函数，将两列数据相加
def add_cols(row):
    return row['A'] + row['B']

# 使用 apply 函数，将函数应用到每一行，得到新的一列 C
df['C'] = df.apply(add_cols, axis=1)

print(df)

输出结果为：

   A  B  C
0  1  4  5
1  2  5  7
2  3  6  9

在这个例子中，我们首先创建了一个 DataFrame `df`，包含两列数据 `A` 和 `B`。然后，我们定义了一个函数 `add_cols`，用于将每一行的 `A` 和 `B` 列相加得到新的一列 `C`。最后，我们使用 `apply` 函数将该函数应用到每一行，得到新的一列 `C`。注意，在这里我们需要指定 `axis=1`，表示函数应用到每一行。