Совокупное количество, где ключ равен значению из другого столбца в Python Pandas

Question

Я пытаюсь составить кумулятивный счет для каждой команды в моем DataFrame, где team = df ['result'] == 'W' .«W» означает победу, поэтому я пытаюсь подсчитать, сколько игр выиграла каждая команда до следующей игры.Это мой код.

df = pd.DataFrame({
'team': ['Inter', 'Barca', 'Psv', 'Totten', 'Psv', 'Barca', 'Inter', 'Totten', 'Totten', 'Psv', 'Inter', 'Barca'],
'result': ['W', 'W', 'L', 'L', 'D', 'W', 'D', 'W', 'W', 'L', 'D', 'D']
})

df['each_played'] = df.groupby('team').cumcount()
df['each_won'] = ???
print(df)

Я успешно подсчитал, сколько игр каждая команда играла до игры, но не могу заставить его работать на df ['each_won'].

Желаемый результат:

     team       result       each_played    each_won
0    Inter      W            0              0
1    Barca      W            0              0
2      Psv      L            0              0
3   Totten      L            0              0
4      Psv      D            1              0
5    Barca      W            1              1
6    Inter      D            1              1
7   Totten      W            1              0
8   Totten      W            2              1
9      Psv      L            2              0
10   Inter      D            2              1
11   Barca      D            2              2

Я новичок в пандах, любая помощь будет принята с благодарностью.

Answer 1

Ваша вторая проблема - это cumsum проблема.Вам понадобится shift и cumsum при вызове GroupBy.apply.

df['each_won'] = (df.result
                    .eq('W')
                    .groupby(df.team)
                    .apply(lambda x: x.shift().cumsum())
                    .fillna(0, downcast='infer'))
df
      team result  each_played each_won
0    Inter      W            0        0
1    Barca      W            0        0
2      Psv      L            0        0
3   Totten      L            0        0
4      Psv      D            1        0
5    Barca      W            1        1
6    Inter      D            1        1
7   Totten      W            1        0
8   Totten      W            2        1
9      Psv      L            2        0
10   Inter      D            2        1
11   Barca      D            2        2