Фильтрация списка строк на основе содержимого

Question

Учитывая список ['a','ab','abc','bac'], я хочу вычислить список со строками, которые содержат 'ab'. То есть результат ['ab','abc']. Как это можно сделать в Python?

Answer 1

Эта простая фильтрация может быть достигнута многими способами с Python. Наилучший подход заключается в использовании «списочных представлений» следующим образом:

>>> lst = ['a', 'ab', 'abc', 'bac']
>>> res = [k for k in lst if 'ab' in k]
>>> res
['ab', 'abc']
>>> 

Другой способ - использовать функцию filter:

>>> filter(lambda k: 'ab' in k, lst)
['ab', 'abc']
>>> 

Answer 2

[x for x in L if 'ab' in x]

Answer 3

# To support matches from the beginning, not any matches:

items = ['a', 'ab', 'abc', 'bac']
prefix = 'ab'

filter(lambda x: x.startswith(prefix), items)

Answer 4

Быстро попробовал это в интерактивной оболочке:

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

Почему это работает? Поскольку оператор in определен для строк, чтобы означать: «является подстрокой».

Кроме того, вы можете рассмотреть возможность записи цикла вместо использования синтаксиса списка , используемого выше:

l = ['a', 'ab', 'abc', 'bac']
result = []
for s in l:
   if 'ab' in s:
       result.append(s)

Answer 5

mylist = ['a', 'ab', 'abc']
assert 'ab' in mylist