Можете ли вы создать новые столбцы из объекта groupby в одну строку, используя панд?

Question

Мне интересно, смогу ли я сделать следующее в одной строке или нужно ли это сделать в две (я из R, откуда знаю, как это сделать за один вызов).Я хочу вычислить среднее значение, которое требует манипулирования столбцами попаданий и битами

import pandas as pd

batting = pd.DataFrame({'playerID': [1, 1, 1, 2, 2, 2],
                        'h': [80, 97, 95, 30, 35, 22],
                        'ab': [400, 410, 390, 150, 170, 145]})

batters = (batting.groupby('playerID')
                  .agg({'h' : 'sum', 'ab' : 'sum'})
                  .reset_index())

batters['ba'] = batters['h']/batters['ab']

Answer 1

eval твой друг .

(batting.groupby('playerID')
        .agg({'h' : 'sum', 'ab' : 'sum'})
        .reset_index()
        .eval('ba = h / ab'))

   playerID    h    ab        ba
0         1  272  1200  0.226667
1         2   87   465  0.187097

Вы можете сократить это до,

batting.groupby('playerID', as_index=False).sum().eval('ba = h / ab')

   playerID    h    ab        ba
0         1  272  1200  0.226667
1         2   87   465  0.187097