快捷導(dǎo)航

Python itertools.product方法代碼實(shí)例

更新時(shí)間：2020年03月27日 10:51:19 投稿：yaominghui

這篇文章主要介紹了Python itertools.product方法代碼實(shí)例,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

itertools.product：類似于求多個(gè)可迭代對(duì)象的笛卡爾積。

使用的形式是：

itertools.product(*iterables, repeat=1),

product(X, repeat=3)等價(jià)于product(X, X, X)。

1. 直接使用時(shí)：分別生成元組，然后合成一個(gè)list

import itertools
aa = itertools.product(['西藏','瀑布','湖水'], ['月色','星空'])
bb = list(aa)  #按照順序生成笛卡爾積，repeat默認(rèn)是1
print(bb)

2. 假設(shè)設(shè)置：repeat=3

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

為什么會(huì)是216呢？

首先，在不設(shè)置 repeat 參數(shù)的時(shí)候，默認(rèn)是1，生成的list長(zhǎng)度時(shí)6 —— 這可以用數(shù)學(xué)的排列組合來(lái)表示，從第一個(gè)參數(shù)['西藏','瀑布','湖水']取出一個(gè)值，有3種可能；從第二個(gè)參數(shù)['月色','星空']取出一個(gè)值，有2種可能；故 3*2=6種結(jié)果。

然后，當(dāng)設(shè)置 repeat=3 時(shí)，也就是說(shuō)將 repeat=1（默認(rèn)）的結(jié)果再重復(fù)2次后（也就是最后一共有3套一樣的第一層結(jié)果）再進(jìn)行排列組合，從第一個(gè)結(jié)果（6種結(jié)果）取出一個(gè)元素的可能有6種，同理，從第二第三個(gè)重復(fù)結(jié)果中取出一個(gè)元素的可能各有6種，于是它們的組合就有6*6*6=216種。

當(dāng)然，也是先合成一個(gè)元組再組成list的。

3. 所以也能理解下面的結(jié)果是27種了吧

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

因?yàn)榈谝粚咏Y(jié)果是3種可能；執(zhí)行該操作3次，就形成了第一層的3套一樣的結(jié)果（每個(gè)結(jié)果各有3種可能），然后再排列組合就是 3*3*3=27種結(jié)果。

4. 如果要從列表中隨機(jī)取出幾個(gè)不重復(fù)的元素的話（原來(lái)的列表本身元素不重復(fù)），可用 random.sample 方法。

import random
random.seed(1)  #設(shè)置隨機(jī)數(shù)種子，可用來(lái)檢測(cè)相同的隨機(jī)數(shù)得到的結(jié)果是否一致
n = 2
aa = random.sample(random_list, n)  #隨機(jī)列表中的n個(gè)元素
print(aa)

也就是從上述27種結(jié)果中，隨機(jī)取出2種，得到：

5. 此方法可以用于生成隨機(jī)的坐標(biāo)

random_list = list(itertools.product(range(1,4), range(1,2)))
print(random_list)
n = 2
aa = random.sample(random_list, n)  #隨機(jī)列表中的n個(gè)元素
print(aa)