Получите массив из объекта itertools.chain

Question

Предположим, у меня есть list_of_numbers = [[1, 2], [3], []], и я хочу гораздо более простой объект списка объектов x = [1, 2, 3].

Следуя логике this related solution, я

list_of_numbers = [[1, 2], [3], []] 
import itertools 
chain = itertools.chain(*list_of_numbers) 

К сожалению, chain не совсем то, что я хочу, потому что (например) работает chain на консоль возвращает <itertools.chain object at 0x7fb535e17790>.

Что такое функция f такой, что если я делаю x = f(chain), а затем введите x на консоли, я получаю [1, 2, 3]?

Обновление: Фактически, в конечном итоге я нуждаюсь в array([1, 2, 3]). Я добавляю строку в комментарии к выбранному ответу, чтобы ответить на это.

Answer 1

list. Если вы делаете list(chain), он должен работать. Но используйте это только для целей отладки, это может быть неэффективным вообще.

Answer 2

Вы можете сделать это с помощью list(chain).

Answer 3

Если ваша конечная цель состоит в том, чтобы получить массив Numpy, то вы должны использовать numpy.fromiter здесь:

>>> import numpy as np 
>>> from itertools import chain 
>>> list_of_numbers = [[1, 2], [3], []] 
>>> np.fromiter(chain(*list_of_numbers), dtype=int) 
array([1, 2, 3]) 
>>> list_of_numbers = [[1, 2]*1000, [3]*1000, []]*1000 
>>> %timeit np.fromiter(chain(*list_of_numbers), dtype=int) 
10 loops, best of 3: 103 ms per loop 
>>> %timeit np.array(list(chain(*list_of_numbers))) 
1 loops, best of 3: 199 ms per loop