首頁 > 後端開發 > Python教學 > Python兩個清單進行全組合的方法是什麼

Python兩個清單進行全組合的方法是什麼

王林
發布: 2023-05-05 20:34:12
轉載
1127 人瀏覽過

什麼是全組合?

兩個元組(a, b)(c, d),則它們的組合有a,c a,d b,c b,d

方式1

透過itertools類別產生一個列表列表,其中包含兩個列表之間的所有潛在組合

import itertools

list1 = [1, 2, 3, 4]
list2 = [5, 6, 7, 8,9]
print(list(itertools.product(list1, list2)))
登入後複製

輸出結果:

[(1, 5), (1, 6), (1, 7), (1, 8), (1, 9), (2, 5), (2, 6), (2, 7), (2, 8), (2, 9), (3, 5), (3, 6), (3, 7), (3, 8), (3, 9), (4, 5), (4, 6), (4, 7), (4, 8), (4, 9)]
登入後複製

方式2

嘗試使用列表產生器建立嵌套列表:

>>> [[[x,y] for x in list1] for y in list2]
[[[1, 3], [2, 3]], [[1, 4], [2, 4]]]
登入後複製

或者,如果需要一行列表,只需刪除括號:

>>> [[x,y] for x in list1 for y in list2]
[[1, 3], [1, 4], [2, 3], [2, 4]]
登入後複製

方式3

repeat第一個列表,permutate第二個列表和zip它們一起

>>> from itertools import permutations, repeat
>>> a = [1, 2, 3]
>>> b = [4, 5, 6]
>>> list(list(zip(r, p)) for (r, p) in zip(repeat(a), permutations(b)))
[[(1, 4), (2, 5), (3, 6)],
 [(1, 4), (2, 6), (3, 5)],
 [(1, 5), (2, 4), (3, 6)],
 [(1, 5), (2, 6), (3, 4)],
 [(1, 6), (2, 4), (3, 5)],
 [(1, 6), (2, 5), (3, 4)]]
登入後複製

編輯:正如Peter Otten所指出的,內部zip和repeat是多餘的。

[list(zip(a, p)) for p in permutations(b)]
登入後複製

以上是Python兩個清單進行全組合的方法是什麼的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:yisu.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板