例如,若要对以'sales”结尾的列进行求和,并将结果放入新列'total”中,可以采用以下代码:
# 定义要求和的列名列表
columns_to_sum = [col for col in df.columns if col.endswith('sales')]
# 对列进行求和,将结果放入新列"total"中
df['total'] = df[columns_to_sum].sum(axis=1)
其中,df.columns
表示数据框的列名列表,col.endswith('sales')
表示当前列名是否以'sales”结尾,columns_to_sum
即为需要求和的列名的列表。df[columns_to_sum].sum(axis=1)
表示对列名在columns_to_sum
中的列沿行方向求和,axis=1
表示沿行方向求和,即对每一行中的所有列进行求和。
最终,将求和结果作为新列total
添加到数据框中。