熊猫数据框如何根据另一列添加列

Question

    |  red  | blue  | winner|
    |:------|:-----:|------:|
0   |70.0   |67.0   |Red    |
1   |74.0   |76.0   |Red    |
2   |75.0   |75.0   |Red    |
3   |63.0   |61.0   |Blue   |
4   |68.0   |72.0   |Blue   |

在 pandas DataFrame 中，只有当它们是获胜者列上显示的获胜者时，我如何才能将红色和蓝色列相加。

在这个例子中

• 红色 = 70 + 74 + 75
• 蓝色 = 61 + 72

Answer 1

import pandas as pd

df = pd.DataFrame(
    data={
        'red' : [70.0, 74.0, 75.0, 63.0, 68.0],
        'blue' : [67.0, 76.0, 75.0, 61.0, 72.0],
        'winner' : ['Red', 'Red', 'Red', 'Blue', 'Blue']
    }
)
print(df['red'][df['winner'] == 'Red'].sum())
print(df['blue'][df['winner'] == 'Blue'].sum())