pandas reset_index após groupby.value_counts ()
Estou tentando agrupar por uma coluna e calcular o valor conta em outra coluna.
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 parece
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
executar agrupamento
grouper = dftest.groupby('A')
df_grouped = grouper['Amt'].value_counts()
que dá
A Amt
1 30 4
20 3
40 2
2 40 3
10 2
Name: Amt, dtype: int64
o que eu quero é manter as duas primeiras linhas de cada grupo
Além disso, fiquei perplexo com um erro ao tentarreset_index
df_grouped.reset_index()
que dá o seguinte erro
df_grouped.reset_index () ValueError: não é possível inserir Amt, já existe