Python itertools.product 详解：笛卡尔积生成与随机采样应用

itertools.product 基础概念与语法

itertools.product 方法主要用于计算多个可迭代对象的笛卡尔积，即生成所有可能的元素组合。

该函数的标准调用形式如下所示：

itertools.product(*iterables, repeat=1)

其中，product(X, repeat=3) 等价于将 X 作为参数重复三次，即 product(X, X, X)。

笛卡尔积生成与 repeat 参数详解

1. 基础用法：直接调用时，函数会生成元组序列，并可转换为列表。

import itertools
aa = itertools.product(['西藏','瀑布','湖水'], ['月色','星空'])
bb = list(aa)  #按照顺序生成笛卡尔积，repeat默认是1
print(bb)

2. 当设置 repeat=3 参数时：

random_list = list(itertools.product(['西藏','瀑布','湖水'], ['月色','星空'], repeat=3))
print(random_list) #此list长度为216

为何结果数量为 216？

首先，若不设置 repeat 参数，其默认值为 1。此时生成的列表长度为 6，这可以通过数学排列组合来解释：从第一个参数 ['西藏','瀑布','湖水'] 中取一个值有 3 种可能，从第二个参数 ['月色','星空'] 中取一个值有 2 种可能，因此总组合数为 3*2=6 种结果。

其次，当设置 repeat=3 时，意味着将默认（repeat=1）的结果重复 2 次，即最终共有 3 套相同的初始结果集进行排列组合。从第一层结果（共 6 种）中取一个元素有 6 种可能，同理，从第二层和第三层重复结果中各取一个元素也各有 6 种可能，因此总组合数为 6*6*6=216 种。

最终结果同样是先合成元组，再组成列表。

3. 基于上述逻辑，可以推导出以下结果为 27 种的情况：

random_list = list(itertools.product(['西藏','瀑布','湖水'], repeat=3))
print(random_list)

这是因为第一层结果有 3 种可能；执行该操作 3 次后，形成了 3 套相同的初始结果（每套各有 3 种可能），再进行排列组合即为 3*3*3=27 种结果。

结合 random.sample 进行随机采样与坐标生成

4. 若需从列表中随机取出几个不重复的元素（前提是原列表本身元素不重复），可使用 random.sample 方法。

import random
random.seed(1)  #设置随机数种子，可用来检测相同的随机数得到的结果是否一致
n = 2
aa = random.sample(random_list, n)  #随机列表中的n个元素
print(aa)

即从上述生成的 27 种结果中，随机选取 2 种，示例输出如下：

5. 此方法也可用于生成随机坐标。

random_list = list(itertools.product(range(1,4), range(1,2)))
print(random_list)
n = 2
aa = random.sample(random_list, n)  #随机列表中的n个元素
print(aa)

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持本站。