Pandas ersetzen Spalte mit Mittelwert für Werte

Question

Ich habe einen Pandas-Datenrahmen und will jeden Wert mit dem Mittelwert dafür ersetzen.

ID X  Y 
    1  a  1 
    2  a  2 
    3  a  3 
    4  b  2 
    5  b  4 

Wie ersetze ich Y-Werte durch Mittelwert Y für jedes eindeutige X?

ID X  Y 
    1  a  2 
    2  a  2 
    3  a  2 
    4  b  3 
    5  b  3 

Answer 1

Verwendung transform:

df['Y'] = df.groupby('X')['Y'].transform('mean') 
print (df) 
    ID X Y 
0 1 a 2 
1 2 a 2 
2 3 a 2 
3 4 b 3 
4 5 b 3 

Für neue Spalte in einer anderen DataFrame Verwendung map mit drop_duplicates:

df1 = pd.DataFrame({'X':['a','a','b']}) 
print (df1) 
    X 
0 a 
1 a 
2 b 

df1['Y'] = df1['X'].map(df.drop_duplicates('X').set_index('X')['Y']) 
print (df1) 
    X Y 
0 a 2 
1 a 2 
2 b 3 

Eine andere Lösung:

df1['Y'] = df1['X'].map(df.groupby('X')['Y'].mean()) 
print (df1) 
    X Y 
0 a 2 
1 a 2 
2 b 3