基于元素字符串的过滤列表

Question

I have a line of code which filters a list if it contains 'ab'

>>> l = ['a', 'ab', 'abc', 'bac']
>>> [x for x in l if 'ab' in x]
['ab', 'abc']
>>>

I now want to say do this if it contains ab or zz, but the below code isn't working

[x for x in l if '|'.join(['ab','zz']) in x]

Answer 1

You can try using a or :

[x for x in l if 'ab' in x or 'zz' in x] 

---

>>> l = ['a', 'ab', 'abc', 'bac', 'zz', 'kkzz']
>>> [x for x in l if 'ab' in x or 'zz' in x]
>>> # ['ab', 'abc', 'zz', 'kkzz']