sprawdzić, czy ciąg w kolumnie dataframe pandy jest na liście

Question

Jeśli mam ramkę jak ten

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

i chcę, aby sprawdzić, czy któryś z tych wierszy zawiera pewną słowo po prostu muszę to zrobić.

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

frame['b'] wyjścia:

True 
False 
True 

Jeśli zdecyduję się zrobić listę

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

jaki sposób mogę sprawdzić, że wiersze zawierają pewną słowo na liście?

Answer 1

Sposób str.contains akceptuje wzorca wyrażenia regularnego:

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

Na liście powinny działać

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

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