Python Pandas Pivot表 - 计数点

Posted 2021-04-19

我在Python中有一个Pivot表的问题。假设我在列表中有以下值：

team_A_id = [1,5,10]

team_A_result = 0

以下数据框：

id             points
3                36
4                0
5                11
7                6
10               23

我如何使用（也许）“for循环”在列表点中找到团队A id并计算它们。输出应该是：

result_team_A = 34

谢谢你的帮助

答案

您正在寻找isin和sum

team_A_id = [1,5,10]
df.loc[df.id.isin(team_A_id),'points'].sum()
Out[136]: 34

另一答案

这将返回团队A的行：

df.iloc[team_A_id]

结果团队A可以通过以下方式获得：

df['points].sum()

TLDR：

df.iloc[team_A_id]['points].sum()