python re.compile () и re.findall ()

Question

Поэтому я пытаюсь напечатать только месяц, и когда я использую:

regex = r'([a-z]+) \d+'
re.findall(regex, 'june 15')

И он печатает: июнь Но когда я пытаюсь сделать то же самое для списка, подобного этому:

regex = re.compile(r'([a-z]+) \d+')
l = ['june 15', 'march 10', 'july 4']
filter(regex.findall, l)

он печатает тот же список, как если бы он не учитывал тот факт, что мне не нужен номер.

Answer 1

Используйте map вместо filter, как в этом примере:

import re

a = ['june 15', 'march 10', 'july 4']
regex = re.compile(r'([a-z]+) \d+')
# Or with a list comprehension
# output = [regex.findall(k) for k in a]
output = list(map(lambda x: regex.findall(x), a))
print(output)

Выход:

[['june'], ['march'], ['july']]

Бонус:

Чтобы сгладить список списков, вы можете сделать:

output = [elm for k in a for elm in regex.findall(k)]
# Or:
# output = list(elm for k in map(lambda x: regex.findall(x), a) for elm in k)

print(output)

Вывод:

['june', 'march', 'july']