Python：如何根據其他條件改變一列的值？

Question

我有一個熊貓數據框。大約有50列和50000行。如果一個colums有

str.contains('certain response') 

我想換另一列響應

'NA' 

我會怎麼做這行？

Answer 1

我相信你需要str.contains的條件，如果需要通過NaN使用np.nan取代：

df.loc[df.column1.str.contains('certain response'), 'column 2'] = np.nan 

或者使用mask，如果True得到NaN S：

df['column 2'] = df['column 2'].mask(df.column1.str.contains('certain response')) 

樣品：

df = pd.DataFrame({'column1':['certain response','certain response 2','aaa'], 
        'column 2':[2,3,4]}) 
df['column 2'] = df['column 2'].mask(df.column1.str.contains('certain response')) 
print (df) 
    column 2    column1 
0  NaN certain response 
1  NaN certain response 2 
2  4.0     aaa