Я пытаюсь группировать столбец и вычислять значения в другом столбце.
import pandas as pd
dftest = pd.DataFrame({'A':[1,1,1,1,1,1,1,1,1,2,2,2,2,2],
'Amt':[20,20,20,30,30,30,30,40, 40,10, 10, 40,40,40]})
print(dftest)
dftest выглядит как
A Amt
0 1 20
1 1 20
2 1 20
3 1 30
4 1 30
5 1 30
6 1 30
7 1 40
8 1 40
9 2 10
10 2 10
11 2 40
12 2 40
13 2 40
выполнить группировку
grouper = dftest.groupby('A')
df_grouped = grouper['Amt'].value_counts()
который дает
A Amt
1 30 4
20 3
40 2
2 40 3
10 2
Name: Amt, dtype: int64
Я хочу сохранить две верхние строки каждой группы
Кроме того, я был озадачен ошибкой при попытке reset_index
df_grouped.reset_index()
который дает следующую ошибку
df_grouped.reset_index() ValueError: не может вставить Amt, уже существует