kiểm tra nếu chuỗi trong cột dataframe gấu trúc có trong danh sách

Question

Nếu tôi có một khung như thế này

frame = pd.DataFrame({'a' : ['the cat is blue', 'the sky is green', 'the dog is black']}) 

và tôi muốn kiểm tra nếu có của những hàng chứa một từ nhất định tôi chỉ phải làm điều này.

frame['b'] = frame.a.str.contains("dog") | frame.a.str.contains("cat") | frame.a.str.contains("fish") 

frame['b'] kết quả đầu ra:

True 
False 
True 

Nếu tôi quyết định để thực hiện một danh sách

mylist =['dog', 'cat', 'fish'] 

thế nào tôi sẽ kiểm tra xem hàng có chứa một từ nào đó trong danh sách?

Answer 1

Phương pháp str.contains chấp nhận một mẫu biểu thức chính quy:

In [11]: pattern = '|'.join(mylist) 

In [12]: pattern 
Out[12]: 'dog|cat|fish' 

In [13]: frame.a.str.contains(pattern) 
Out[13]: 
0  True 
1 False 
2  True 
Name: a, dtype: bool 

Answer 2

Đối với danh sách nên làm việc

print frame[frame['a'].isin(mylist)]  

http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.isin.html