基于条件用NaN替换数据帧值同时保持d的形状

Question

我是Python新手。我肯定我问的是一个无关紧要的问题，但在做了一点googlefoo之后，不幸的是我还没有找到解决办法。所以就这样： 如果我有这样的数据帧：

raw_data = {'first_name': ['Jason', 'Molly', 'Tina', 'Jake', 'Amy'], 
    'last_name': ['Miller', 'Jacobson', 'Ali', 'Milner', 'Cooze'], 
    'age': [42, 52, 36, 24, 73], 
    'preTestScore': [-999, -999, -999, 2, 1],
    'postTestScore': [2, 2, -999, 2, -999]}
df = pd.DataFrame(raw_data, columns = ['first_name', 'last_name', 'age', 'preTestScore', 'postTestScore'])

如何执行以下操作： 如果preTestScore=-999，那么用NaN替换preTestScore和postestscore？你知道吗

我可以使用测向替换(-999, np.nan公司)但这需要跨两列有条件地删除。你知道吗

谢谢你的好意

Answer 1

将^{}与boolen mask一起使用，并将列列表设置为NaN：

df.loc[df['preTestScore'] == -999, ['preTestScore','postTestScore']] = np.nan
print (df)

      first_name last_name  age  preTestScore  postTestScore
0      Jason    Miller   42           NaN            NaN
1      Molly  Jacobson   52           NaN            NaN
2       Tina       Ali   36           NaN            NaN
3       Jake    Milner   24           2.0            2.0
4        Amy     Cooze   73           1.0         -999.0

细节：

print (df['preTestScore'] == -999)
0     True
1     True
2     True
3    False
4    False
Name: preTestScore, dtype: bool