在pandas中,可以使用apply()函数将函数应用于数据帧的每一行,其中函数的输入是列表形式的行中的元素。
具体步骤如下:
以下是一个示例代码:
import pandas as pd
# 定义一个函数,对列表中的元素进行求和
def sum_row(row):
return sum(row)
# 创建一个数据帧
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
# 使用apply()函数将sum_row函数应用于每一行
df['Sum'] = df.apply(sum_row, axis=1)
print(df)
输出结果为:
A B C Sum
0 1 4 7 12
1 2 5 8 15
2 3 6 9 18
在这个示例中,我们定义了一个sum_row函数,用于对列表中的元素求和。然后使用apply()函数将该函数应用于数据帧的每一行,并将结果存储在新的一列'Sum'中。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云