pandas reset_index después de groupby.value_counts ()
Estoy tratando de agrupar por una columna y calcular el recuento de valores en otra columna.
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
realizar agrupamiento
grouper = dftest.groupby('A')
df_grouped = grouper['Amt'].value_counts()
lo que da
A Amt
1 30 4
20 3
40 2
2 40 3
10 2
Name: Amt, dtype: int64
lo que quiero es mantener las dos primeras filas de cada grupo
Además, estaba perplejo por un error cuando intentéreset_index
df_grouped.reset_index()
que da el siguiente error
df_grouped.reset_index () ValueError: no se puede insertar Amt, ya existe