我有下一个pandas DataFrame:
a b c
1 1 5.0
1 1 None
1 1 4.0
1 2 1.0
1 2 1.0
1 2 4.0
2 1 3.0
2 1 2.0
2 1 None
2 2 3.0
2 2 4.0
我想要替换None
,而不是替换列的意思。我想选择所有行,其中a
和b
中的值相似,如果c
具有None
- 所选行中的值,请仅使用{替换它们{1}} - 所选行的平均值。像(这段代码不起作用):
c
应该得到我的输出:
df[df[('a'==1) & ('b'==1)]]['c'].fillna(df[df[('a'==1) & ('b'==1)]]['c'].mean())
答案 0 :(得分:0)
您需要按条件筛选c
的值,并指定列c
:
mask = (df['a']==1) & (df['b']==1)
mean = df.loc[mask, 'c'].mean()
df.loc[mask, 'c'] = df.loc[mask, 'c'].fillna(mean)
或使用mask
替换条件:
df['c'] = df['c'].mask(mask, df['c'].fillna(mean))
#similar
#df['c'] = np.where(mask, df['c'].fillna(mean), df['c'])
print (df)
a b c
0 1 1 5.0
1 1 1 4.5
2 1 1 4.0
3 1 2 1.0
4 1 2 1.0
5 1 2 4.0
6 2 1 3.0
7 2 1 2.0
8 2 1 NaN
9 2 2 3.0
10 2 2 4.0