可以使用apply函数来对同一数据框的多个列应用某个函数。例如,下面的代码将df的列A和列B加以求和,并将结果存储在新列C中:
import pandas as pd
df = pd.DataFrame({'A': [1, 3, 5], 'B': [2, 4, 6]})
df['C'] = df[['A', 'B']].apply(lambda x: x['A'] + x['B'], axis=1)
print(df)
输出:
A B C
0 1 2 3
1 3 4 7
2 5 6 11
其中,lambda表达式指定了加法操作,axis参数指定了对列进行操作。可以根据实际需求编写不同的lambda表达式。