Я использую itertools.chain для "сглаживания" списка списков таким образом:
uniqueCrossTabs = list(itertools.chain(*uniqueCrossTabs))
как это отличается от высказывания:
uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))
Я использую itertools.chain для "сглаживания" списка списков таким образом:
uniqueCrossTabs = list(itertools.chain(*uniqueCrossTabs))
как это отличается от высказывания:
uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))
*
- это оператор "splat": он принимает список как входной сигнал и расширяет его в фактические позиционные аргументы в вызове функции.
Итак, если uniqueCrossTabs
было [ [ 1, 2 ], [ 3, 4 ] ]
, то itertools.chain(*uniqueCrossTabs)
совпадает с itertools.chain([ 1, 2 ], [ 3, 4 ])
Это, очевидно, отличается от передачи только uniqueCrossTabs
. В вашем случае у вас есть список списков, которые вы хотите сгладить; то, что itertools.chain()
делает, возвращает итератор по конкатенации всех позиционных аргументов, которые вы передаете ему, где каждый позиционный аргумент итерируется сам по себе.
Другими словами, вы хотите передать каждый список в uniqueCrossTabs
в качестве аргумента в chain()
, который свяжет их вместе, но у вас нет списков в разных переменных, поэтому вы используете *
оператора, чтобы развернуть список списков в несколько аргументов списка.
Как отметил Йохен Ритцел в комментариях, chain.from_iterable()
лучше подходит для этой операции, так как он предполагает одноименный итерабель для итераций. Тогда ваш код будет просто:
uniqueCrossTabs = list(itertools.chain.from_iterable(uniqueCrossTabs))
Он разбивает последовательность на отдельные аргументы для вызова функции.
>>> def foo(a, b=None, c=None):
... print a, b, c
...
>>> foo([1, 2, 3])
[1, 2, 3] None None
>>> foo(*[1, 2, 3])
1 2 3
>>> def bar(*a):
... print a
...
>>> bar([1, 2, 3])
([1, 2, 3],)
>>> bar(*[1, 2, 3])
(1, 2, 3)
Просто альтернативный способ объяснить концепцию/использовать ее.
import random
def arbitrary():
return [x for x in range(1, random.randint(3,10))]
a, b, *rest = arbitrary()
# a = 1
# b = 2
# rest = [3,4,5]