使用df.loc []和在

Question

我試圖做下面的代碼：

df.loc[df['Column'] in list]=1 
df.loc[df['Column'] not in list]=2 

但我有著名的錯誤

一個系列的真值是不明確的。使用a.empty，a.bool（），a.item（），a.any（）或a.all（）。

因此，從我瞭解的這個錯誤我已經得到，它是與邏輯鏈接。我想知道的是有沒有一個特定的錯字來做我想要做的事情，或者我需要擴展我的整個列表（在我的情況下這將是相當醜陋和長）：

df.loc[df[('Column']=='a') & (df['Column']=='b')]=1 
df.loc[df[('Column']!='a') & (df['Column']!='b')]=2 

Answer 1

我會用np.where。但是，您需要列名稱。

df['new_column'] = np.where(df['Column'].isin(lst), 1, 2) 

考慮名單lst和數據幀df

lst = [1, 2, 3] 
df = pd.DataFrame(dict(Column=[0, 1, 2, 3, 4])) 

然後

df['new_column'] = np.where(df['Column'].isin(lst), 1, 2) 

print(df) 

    Column new_column 
0  0   2 
1  1   1 
2  2   1 
3  3   1 
4  4   2 

Answer 2

你正在尋找的功能是

df.loc[df['Column'].isin(list)] = 1