Python 应用itertools.product到一个列表的列表的元素

itertools 是Python标准库中的一个模块，提供了一组用于高效迭代和组合可迭代对象的工具。由于它是Python标准库的一部分，所以不需要进行任何额外的安装。它提供了各种函数，可以以不同的方式操作、组合和迭代可迭代对象。

itertools.product() 函数与 itertools 模块相关，是生成多个 可迭代对象 的笛卡尔积的强大工具。它接受一个或多个 可迭代对象 作为输入，并返回一个生成所有可能组合的元素的元组的迭代器。

itertools.product 返回一个迭代器；换句话说，它会在请求时即时生成元组。这在处理大型输入的 可迭代对象 或者组合数非常大时非常节省内存。

语法

使用 itertools.product() 函数的语法如下：

itertools.product(*iterables, repeat=1)

下面是关于使用 itertools.product() 生成颜色和尺码的所有可能组合的示例。生成的迭代器 product 包含代表每个组合的元组，我们遍历它并打印每个元组。

import itertools
colors = ['red', 'green', 'blue']
sizes = ['S', 'M', 'L']
product = itertools.product(colors, sizes)
for item in product:
   print(item)

输出

('red', 'S')
('red', 'M')
('red', 'L')
('green', 'S')
('green', 'M')
('green', 'L')
('blue', 'S')
('blue', 'M')
('blue', 'L')

示例

在这个例子中，我们尝试使用itertools.product()函数计算字母列表和数字列表的笛卡尔积。生成的迭代器产生由字母列表中的一个元素和数字列表中的一个元素组成的所有可能组合的元组。然后使用循环打印出这些元组。

import itertools
letters = ['a', 'b']
numbers = [1, 2, 3]
result = itertools.product(letters, numbers)
for item in result:
   print(item)

输出

('a', 1)
('a', 2)
('a', 3)
('b', 1)
('b', 2)
('b', 3)

示例

在之前的例子中，我们计算了数字和字母的笛卡尔积，重复次数为一次。在这个例子中，重复参数被设置为3，这会导致笛卡尔积重复三次，生成的迭代器会生成由数字列表中所有可能的三个元素组成的元组，包括重复元素。

import itertools
letters = ['a', 'b']
numbers = [1, 2, 3]
result = itertools.product(letters, numbers, repeat = 2)
for item in result:
   print(item)

结果

('a', 1, 'a', 1)
('a', 1, 'a', 2)
('a', 1, 'a', 3)
('a', 1, 'b', 1)
('a', 1, 'b', 2)
('a', 1, 'b', 3)
('a', 2, 'a', 1)
('a', 2, 'a', 2)
('a', 2, 'a', 3)
('a', 2, 'b', 1)
('a', 2, 'b', 2)
('a', 2, 'b', 3)
('a', 3, 'a', 1)
('a', 3, 'a', 2)
('a', 3, 'a', 3)
('a', 3, 'b', 1)
('a', 3, 'b', 2)
('a', 3, 'b', 3)
('b', 1, 'a', 1)
('b', 1, 'a', 2)
('b', 1, 'a', 3)
('b', 1, 'b', 1)
('b', 1, 'b', 2)
('b', 1, 'b', 3)
('b', 2, 'a', 1)
('b', 2, 'a', 2)
('b', 2, 'a', 3)
('b', 2, 'b', 1)
('b', 2, 'b', 2)
('b', 2, 'b', 3)
('b', 3, 'a', 1)
('b', 3, 'a', 2)
('b', 3, 'a', 3)
('b', 3, 'b', 1)
('b', 3, 'b', 2)
('b', 3, 'b', 3)