pandas. Apply多列

在pandas中，apply函数可以用于同时应用于多列数据。你可以使用apply函数在每一行或每一列上执行自定义函数。

如果你想要在多列上应用同一个函数，可以使用apply函数的axis参数来指定轴的方向。默认情况下，axis=0表示在每一列上应用函数，axis=1表示在每一行上应用函数。

以下是一个示例，演示如何在多列上应用apply函数：

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3],
        'B': [4, 5, 6],
        'C': [7, 8, 9]}

df = pd.DataFrame(data)

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

# 在多列上应用函数
df['D'] = df.apply(sum_two_columns, axis=1)

print(df)

输出结果如下：

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

在这个示例中，我们创建了一个包含三列的DataFrame，并定义了一个函数`sum_two_columns`，它将列'A'和列'B'的值相加。然后，我们使用apply函数将这个函数应用到每一行上，并将结果存储在新的列'D'中。

希望这个示例能帮助你理解如何在pandas中使用apply函数在多列上应用自定义函数。