創建使用兩列之間的差異在大熊貓

Question

這是一個新的數據幀是數據幀的一個子集：

index id drug sentences  SS1 SS2 
1  2 lex  very bad  0  1 
2  3 gym  very nice  1  1 
3  7 effex hard   1  0 
4  8 cymba poor   1  1 

我想找到行時，SS1和SS2是不同的，然後創建一個新的數據幀基於此。輸出應該是這樣的：

index id drug sentences  SS1 SS2 
1  2 lex  very bad  0  1 
3  7 effex hard   1  0 

這是我的代碼：

df [['index','id', 'drug', 'sentences', 'SS1', 'SS2' ]] = np.where(df.SS1 != df.SS2) 

但它有以下錯誤：ValueError: Must have equal len keys and value when setting with an ndarray

什麼建議嗎？

Answer 1

一種方式可以是以下幾點：

df_new = df[df.SS1 != df.SS2] 
print(df_new) 

輸出：

index id drug sentences SS1 SS2 
0  1 2 lex very bad 0 1 
2  3 7 effex  hard 1 0 

使用where：

df_new = df.where(df.SS1 != df.SS2).dropna() 
print(df_new)