Как применять itertools.продукт к элементам списка списков?

у меня есть список массивов и хотелось бы получить декартово произведение элементов в массивах.

Я буду использовать пример, чтобы сделать более конкретным...

itertools.продукт, кажется, делает трюк, но я застрял в небольшой детали.

arrays = [(-1,+1), (-2,+2), (-3,+3)];

если я делаю

cp = list(itertools.product(arrays));

Я 

cp = cp0 = [((-1, 1),), ((-2, 2),), ((-3, 3),)]

но то, что я хочу сделать-это

cp1 = [(-1,-2,-3), (-1,-2,+3), (-1,+2,-3), (-1,+2,+3), ..., (+1,+2,-3), (+1,+2,+3)].

Я пробовал несколько разных вещей:

cp = list(itertools.product(itertools.islice(arrays, len(arrays))));
cp = list(itertools.product(iter(arrays, len(arrays))));

они все дал мне cp0 вместо cp1.

какие идеи?

спасибо заранее.

32
cartesian-product itertools python

2 ответов

>>> list(itertools.product(*arrays))
[(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (1, 2, -3), (1, 2, 3)]

Это будет кормить все пары в качестве отдельных аргументов в product, которая даст вам декартово произведение их.

причина, по которой ваша версия не работает, заключается в том, что вы даете product только один аргумент. Запрос декартового произведения одного списка является тривиальным случаем и возвращает список, содержащий только один элемент (список, заданный как аргумент).

39
>>> arrays = [(-1,+1), (-2,+2), (-3,+3)]
>>> list(itertools.product(*arrays))
[(-1, -2, -3), (-1, -2, 3), (-1, 2, -3), (-1, 2, 3), (1, -2, -3), (1, -2, 3), (1, 2, -3), (1, 2, 3)]
35