python - 如何使用 Pandas 计算列中的字符串？

Question

我有以下数据集。

game_id h_abbr a_abbr 
0001    WSH    TOR 
0002    ANA    TOR 
0003    TOR    MIN

我想计算到目前为止每支球队打了多少场比赛。输出应该是这样的

game_id h_abbr a_abbr ht_game_no at_game_no
0001    WSH    TOR    1          1
0002    ANA    TOR    1          2
0003    TOR    MIN    3          1

Answer 1

使用 DataFrame.set_index 与 DataFrame.stack 对于 MultiIndex Series ，然后使用 GroupBy.cumcount 对于计数器，由 Series.unstack reshape , 添加 rename并通过 DataFrame.join 添加到原始 DataFrame :

s = df.set_index('game_id')[['h_abbr','a_abbr']].stack()

d = {'h_abbr':'ht_game_no','a_abbr':'at_game_no'}
df = df.join(s.groupby(s).cumcount().add(1).unstack().rename(columns=d), on='game_id')
print (df)
   game_id h_abbr a_abbr  ht_game_no  at_game_no
0        1    WSH    TOR           1           1
1        2    ANA    TOR           1           2
2        3    TOR    MIN           3           1