Вернуть список слов из списка строк с регулярным выражением

>>> import re
>>> from itertools import chain
>>> lines = ["hello world", "second line", "third line"]
>>> words = chain(*[re.split(r'\s+', line) for line in lines])

Это даст вам итератор, который можно использовать для перебора всех слов:

>>> for word in words:
...    print(word)
... 
hello
world
second
line
third
line

Создание списка вместо итератора — это просто обертывание итератора вызовом list:

>>> words = list(chain(*[re.split(r'\s+', line) for line in lines]))

Причина, по которой вы получаете список списков, заключается в том, что re.split() возвращает список, который затем «добавляется» к выводу понимания списка.

Непонятно, почему вы это используете (или, возможно, просто плохой пример), но если вы можете получить полное содержимое (все строки) в виде строки, которую вы можете просто сделать

words = re.split(r'\s+', lines)

если строки являются произведением:

open('filename').readlines()

использовать

open('filename').read()

вместо.

Вы всегда можете сделать это:

words = []
for line in lines:
  words.extend(re.split('\\s+',line))

Это не так элегантно, как однострочное понимание списка, но оно выполняет свою работу.

Просто наткнулся на этот старый вопрос, и я думаю, что у меня есть лучшее решение. Обычно, если вы хотите вложить понимание списка («добавить» каждый список), вы думаете в обратном направлении (не похоже на цикл). Это не то, что вы хотите:

>>> import re
>>> lines = ["hello world", "second line", "third line"]
>>> [[word for word in re.split(r'\s+', line)] for line in lines]
[['hello', 'world'], ['second', 'line'], ['third', 'line']]

Однако, если вы хотите «расширить» вместо «добавления» списки, которые вы создаете, просто исключите дополнительный набор квадратных скобок и переверните циклы for (вернув их в «правильный» порядок).

>>> [word for line in lines for word in re.split(r'\s+', line)]
['hello', 'world', 'second', 'line', 'third', 'line']

Это кажется мне более питоническим решением, поскольку оно основано на логике обработки списков, а не на какой-то случайной встроенной функции. Каждый программист должен знать, как это сделать (особенно те, кто пытается изучать Лисп!)