python - 使用 pandas 的棘手聚合

Question

我是 pandas 的新手，并且好奇这是否太棘手而无法完成使用它。

示例输入：

time       person   game_id   won
-----------------------------------
12:34:01   John     3         False
12:34:04   Ringo    2         True
12:35:05   John     3         False
12:36:01   John     3         True
12:36:12   Ringo    3         True
12:36:41   Paul     4         False
12:37:01   George   2         False
12:37:41   George   2         False

它显示了随着时间的推移，许多人玩了许多游戏。获胜的列指示该人当时是否获胜。

我想要的输出是有多少人每场比赛至少赢了一次。但也有多少人玩了游戏却从未赢过。

示例输出：

game_id   won     count
-----------------------
2         True    1
          False   1
3         True    2
          False   0
4         True    0
          False   1

Answer 1

或多或少@DSM 在说什么：

In [3]: grouped = df.groupby('game_id')

In [4]: won = grouped.won.sum()

In [5]: DataFrame({True: won, False: grouped.person.nunique() - won}).stack()
Out[5]: 
game_id       
2        False    1
         True     1
3        False    0
         True     2
4        False    1
         True     0
dtype: float64

Answer 2

s1 = (~df.groupby(["game_id", "person"]).won.agg(np.any)).groupby(level=0).sum()
s2 = df.groupby(["game_id", "won"]).person.agg(lambda s:s.nunique())
df2 = s2.unstack()
df2[False] = s1
df2.fillna(0).unstack().swaplevel(0, 1).sort_index()