如何使用Python Pandas计算列总和、值占比并基于结果创建新列？

完整可运行代码 #

import pandas as pd

# 构造示例DataFrame
df = pd.DataFrame({
    'col1': [10, 20, 30],
    'col2': [2, 2, 1]
})

# 按规则新增4个列
df['col3'] = df['col1'].sum()
df['col4'] = (df['col1'] / df['col3']).round(2)
df['col5'] = df['col2'].sum()
df['col6'] = (df['col2'] / df['col5']).round(2)

# 打印结果
print(df)

实现说明 #

• 列求和得到的是标量值，Pandas会自动将标量广播到所有行，无需额外循环处理
• 示例中第一行col4的0.16为近似值，标准四舍五入后结果为0.17，如需完全匹配示例的截断效果，可将round(2)替换为apply(lambda x: int(x*100)/100)实现保留两位小数截断。

运行输出结果 #

col1  col2  col3  col4  col5  col6
0    10     2    60  0.17     5   0.4
1    20     2    60  0.33     5   0.4
2    30     1    60  0.50     5   0.2

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