Используйте список для фильтрации другого списка в Python

Question

У меня есть список:

data_list = ['a.1','b.2','c.3'] 

И я хочу, чтобы получить только строки, которые начинаются со строками из другого списка:

test_list = ['a.','c.'] 

a.1 и c.3 должны быть вернулся.

Я полагаю, я мог бы использовать двойной для цикла:

for data in data_list: 
    for test in test_list: 
     if data.startswith(test): 
      # do something with item 

мне было интересно, если есть что-то более элегантное и, возможно, более peformant.

Answer 1

str.startswith может также принимать кортеж (но не список) префиксов:

test_tuple=tuple(test_list) 
for data in data_list: 
    if data.startswith(test_tuple): 
     ... 

что означает простой список понимание даст вам отфильтрованный список:

matching_strings = [ x for x in data_list if x.startswith(test_tuple) ] 

или вызов filter:

import operator 
f = operator.methodcaller('startswith', tuple(test_list)) 
matching_strings = filter(f, test_list) 

Answer 2

Попробуйте следующее:

for data in data_list: 
    if any(data.startswith(test) for test in test_list): 
     # do something 

any() является встроенным, которая принимает итератор и возвращает True первого значение из итератора, что Ие правда, в противном случае возвращает False. В моем примере я использую выражение генератора, а не строю список (который был бы расточительным).

Answer 3

Отъезд filter и any в документах python.

>>> data_list = ['a.1','b.2','c.3'] 
>>> test_list = ['a.','c.'] 
>>> new_list = filter(lambda x: any(x.startswith(t) for t in test_list), data_list) 
>>> new_list 
['a.1', 'c.3'] 

После этого вы можете делать все, что вы хотите с материалом в вашем new_list.

Как @Chepner указывает, вы также можете поставить кортеж строк для startswith, так что выше, также может быть написано:

>>> data_list = ['a.1','b.2','c.3'] 
>>> test_tuple = ('a.','c.') 
>>> new_list = filter(lambda x: x.startswith(test_tuple), data_list) 
>>> new_list 
['a.1', 'c.3'] 

Answer 4

Просто используйте filter с lambda function и startswith:

data_list = ['a.1','b.2','c.3'] 
test_list = ('a.','c.') 

result = filter(lambda x: x.startswith(test_list), data_list) 

print(list(result)) 

Выходные:

['a.1', 'c.3'] 

Answer 5

В качестве альтернативы, вспыхивают регулярные выражения

import re 
# build a pattern that matches any of the strings we are interested in 
pattern = re.compile('|'.join(map(re.escape, test_list))) 
# filter by matches 
print filter(pattern.match, data_list) 

Это, вероятно, максимально возможно в C и может быть более эффективным, чем другие решения. Это может быть немного сложно для непосвященных, чтобы следовать, хотя.