Question

我的df包含许多列。我只想根据条件用NaN替换A和B列中的所有值。另外，除了列C和D之外，我想将相同条件应用于另一个df。到目前为止，我的搜索返回的方法适用于单个列。

到目前为止我的尝试。
仅在A和B列上：

df[df.loc[:, df.columns['A','B']] < (0.1 * 500)] = np.nan

除了列A和列B：

df[df.loc[:, df.columns != ['A','B']] < (0.1 * 500)] = np.nan

这些代码不起作用。

Answer 1

我认为您需要DataFrame.mask：

df = pd.DataFrame({
         'A':[4,5,4,5,5,4],
         'B':[7,8,9,4,2,3],
         'C':[1,3,5,7,1,0],
         'D':[5,3,6,9,2,4],

}) * 10

c = ['A','B']
df[c] = df[c].mask(df[c] < (0.1 * 500))
print (df)
      A     B   C   D
0   NaN  70.0  10  50
1  50.0  80.0  30  30
2   NaN  90.0  50  60
3  50.0   NaN  70  90
4  50.0   NaN  10  20
5   NaN   NaN   0  40

c1 = df.columns.difference(c)
df[c1] = df[c1].mask(df[c1] < (0.1 * 500))

print (df)
      A     B     C     D
0   NaN  70.0   NaN  50.0
1  50.0  80.0   NaN   NaN
2   NaN  90.0  50.0  60.0
3  50.0   NaN  70.0  90.0
4  50.0   NaN   NaN   NaN
5   NaN   NaN   NaN   NaN

根据条件用多列上的nan替换值

1 个答案: