python - check if string in pandas dataframe column is in list -
if have frame this
frame = pd.dataframe({'a' : ['the cat blue', 'the sky green', 'the dog black']}) and want check if of rows contain word have this.
frame['b'] = frame.a.str.contains("dog") | frame.a.str.contains("cat") | frame.a.str.contains("fish") frame['b'] outputs:
true false true if decide make list
mylist =['dog', 'cat', 'fish'] how check rows contain word in list?
the str.contains method accepts regular expression pattern:
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
Comments
Post a Comment