itertools.product:类似于求多个可迭代对象的笛卡尔积。
使用的形式是:
itertools.product(*iterables, repeat=1),
product(X, repeat=3)等价于product(X, X, X)。
1. 直接使用时:分别生成元组,然后合成一个list
代码语言:javascript复制import itertools
aa = itertools.product(['西藏','瀑布','湖水'], ['月色','星空'])
bb = list(aa) #按照顺序生成笛卡尔积,repeat默认是1
print(bb)
2. 假设设置:repeat=3
代码语言:javascript复制random_list = list(itertools.product(['西藏','瀑布','湖水'], ['月色','星空'], repeat=3))
print(random_list) #此list长度为216
为什么会是216呢?
首先,在不设置 repeat 参数的时候,默认是1,生成的list长度时6 —— 这可以用数学的排列组合来表示,从第一个参数[‘西藏’,’瀑布’,’湖水’]取出一个值,有3种可能;从第二个参数[‘月色’,’星空’]取出一个值,有2种可能;故 3*2=6种结果。
然后,当设置 repeat=3 时,也就是说将 repeat=1(默认)的结果再重复2次后(也就是最后一共有3套一样的第一层结果)再进行排列组合,从第一个结果(6种结果)取出一个元素的可能有6种,同理,从第二第三个重复结果中取出一个元素的可能各有6种,于是它们的组合就有6*6*6=216种。
当然,也是先合成一个元组再组成list的。
3. 所以也能理解下面的结果是27种了吧
代码语言:javascript复制random_list = list(itertools.product(['西藏','瀑布','湖水'], repeat=3))
print(random_list)
因为第一层结果是3种可能;执行该操作3次,就形成了第一层的3套一样的结果(每个结果各有3种可能),然后再排列组合就是 3*3*3=27种结果。
4. 如果要从列表中随机取出几个不重复的元素的话(原来的列表本身元素不重复),可用 random.sample 方法。
代码语言:javascript复制import random
random.seed(1) #设置随机数种子,可用来检测相同的随机数得到的结果是否一致
n = 2
aa = random.sample(random_list, n) #随机列表中的n个元素
print(aa)
也就是从上述27种结果中,随机取出2种,得到:
5. 此方法可以用于生成随机的坐标
代码语言:javascript复制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)
以上就是本文的全部内容,希望对大家的学习有所帮助。