Pandas DataFrame Groupby Manipulation

Question

У меня есть список транзакций во фрейме данных, и я хочу сгруппировать по Symbols и взять сумму одного из столбцов.Кроме того, я хочу первый экземпляр этого столбца (на символ).

Мой код:

local_filename= 'C:\Users\\nshah\Desktop\Naman\TEMPLATE.xlsx'
data_from_local_file = pd.read_excel(local_filename, error_bad_lines=False, sheet_name='JP_A')
data_from_local_file = data_from_local_file[['Symbol','Security Name', 'Counterparty', 'Manager', 'Rate', 'LocatedAmt']]

data_grouped = data_from_local_file.groupby(['Symbol'])

pivoted  = data_grouped['LocatedAmt'].sum().reset_index()

Далее я хочу первый экземпляр, скажем, с той же скоростьюсимвол.

Заранее спасибо!

Answer 1

Вы можете получить сумму и первый наблюдаемый экземпляр следующим образом:

data_grouped = data_from_local_file.groupby(['Symbol'], as_index=False).agg({'LocatedAmt':[sum, 'first']})

Чтобы выполнить это для всех столбцов, вы можете передать функцию agg во все столбцы:

all_cols = ['Symbol','Security Name', 'Counterparty', 'Manager', 'Rate', 'LocatedAmt']
data_grouped_all = data_from_local_file.groupby(['Symbol'], as_index=False)[all_cols].agg([sum, 'first'])