熊貓：根據字段中包含的字段進行過濾

Question

我有一個DataFrame df和一組user_ids：set。我如何選擇只包含集合中用戶的df。像：

df[df.user_id in user_ids] 

現在的工作，那是因爲：

'Series' objects are mutable, thus they cannot be hashed 

Answer 1

可以使用isin爲boolean mask與boolean indexing過濾：

df = pd.DataFrame({'user_id':[1,2,3], 
        'B':[4,5,6]}) 

print (df) 
    B user_id 
0 4  1 
1 5  2 
2 6  3 

user_ids = set([1,2]) 

print (df.user_id.isin(user_ids)) 
0  True 
1  True 
2 False 
Name: user_id, dtype: bool 

df1 = df[df.user_id.isin(user_ids)] 
print (df1) 
    B user_id 
0 4  1 
1 5  2