J'essaie de trouver le coût mensuel moyen par user_id mais je ne peux obtenir que le coût moyen par utilisateur ou le coût mensuel par utilisateur.
Parce que je regroupe par utilisateur et par mois, il n'y a aucun moyen d'obtenir la moyenne du deuxième groupby (mois) à moins que je ne transforme la sortie groupby en autre chose.
Voici mon df:
df = { 'id' : pd.Series([1,1,1,1,2,2,2,2]),
'cost' : pd.Series([10,20,30,40,50,60,70,80]),
'mth': pd.Series([3,3,4,5,3,4,4,5])}
cost id mth
0 10 1 3
1 20 1 3
2 30 1 4
3 40 1 5
4 50 2 3
5 60 2 4
6 70 2 4
7 80 2 5
Je peux obtenir une somme mensuelle mais je veux la moyenne des mois pour chaque user_id.
df.groupby(['id','mth'])['cost'].sum()
id mth
1 3 30
4 30
5 40
2 3 50
4 130
5 80
je veux quelque chose comme ça:
id average_monthly
1 (30+30+40)/3
2 (50+130+80)/3
La réinitialisation de l'index devrait fonctionner. Essaye ça:
In [19]: df.groupby(['id', 'mth']).sum().reset_index().groupby('id').mean()
Out[19]:
mth cost
id
1 4.0 33.333333
2 4.0 86.666667
Vous pouvez simplement déposer mth
si vous le souhaitez. La logique est qu'après la partie sum
, vous avez ceci:
In [20]: df.groupby(['id', 'mth']).sum()
Out[20]:
cost
id mth
1 3 30
4 30
5 40
2 3 50
4 130
5 80
La réinitialisation de l'indice à ce stade vous donnera des mois uniques.
In [21]: df.groupby(['id', 'mth']).sum().reset_index()
Out[21]:
id mth cost
0 1 3 30
1 1 4 30
2 1 5 40
3 2 3 50
4 2 4 130
5 2 5 80
Il suffit de le regrouper à nouveau, cette fois en utilisant mean
au lieu de sum
. Cela devrait vous donner les moyennes.
Faites-nous savoir si cela vous aide.