Учитывая список ['a','ab','abc','bac'], я хочу вычислить список со строками, которые 'ab'в них есть. Т.е. результат есть ['ab','abc']. Как это можно сделать в Python?
Ответы:
Эта простая фильтрация может быть достигнута разными способами с помощью Python. Наилучший подход - использовать "понимание списка" следующим образом:
>>> lst = ['a', 'ab', 'abc', 'bac']
>>> [k for k in lst if 'ab' in k]
['ab', 'abc']
Другой способ - использовать filterфункцию. В Python 2:
>>> filter(lambda k: 'ab' in k, lst)
['ab', 'abc']
В Python 3 он возвращает итератор вместо списка, но вы можете привести его:
>>> list(filter(lambda k: 'ab' in k, lst))
['ab', 'abc']
Хотя лучше использовать понимание.
lambdaидеальный компаньон filterв этом случае? Я считаю, что писать отдельную функцию только для проверки того, abвходит ли она в данный список, - это излишество. Итак, мы пишем более общую функцию, которая в основном является оболочкой для inоператора. Как бы вы использовали filterболее понятный способ без lambdaздесь?
[x for x in L if 'ab' in x]
# To support matches from the beginning, not any matches:
items = ['a', 'ab', 'abc', 'bac']
prefix = 'ab'
filter(lambda x: x.startswith(prefix), items)
Быстро опробовал это в интерактивной оболочке:
>>> 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)