Python按条件对DataFrame指定列求和并生成新列方法问询

核心逻辑 #

• 先筛选出Price列中所有大于-1的数值，计算这些数值的总和
• 新增Total Sum列，默认赋值为空值
• 仅给Total Sum列的首行填入计算得到的总和

完整可运行代码 #

import pandas as pd

# 构造示例输入DataFrame
data = {'Price': [12,14,15,10,2,4,-5,-4,-3,-5,16,15]}
df = pd.DataFrame(data)

# 计算符合条件的总和
valid_sum = df[df['Price'] > -1]['Price'].sum()

# 新增列，仅首行展示总和
df['Total Sum'] = ''
# 用df.index[0]适配索引非从0开始的场景
df.loc[df.index[0], 'Total Sum'] = valid_sum

# 输出结果
print(df)

运行输出结果 #

Price Total Sum
0     12        88
1     14          
2     15          
3     10          
4      2          
5      4          
6     -5          
7     -4          
8     -3          
9     -5          
10    16          
11    15          

内容的提问来源于stack exchange，提问作者sirimiri