關(guān)于Python兩個列表進行全組合操作的三種方式
什么是全組合?
兩個元組 (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)))
輸出結(jié)果:
[(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
嘗試使用列表生成器創(chuàng)建嵌套列表:
>>> [[[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所指出的,內(nèi)部zip和repeat是多余的。
[list(zip(a, p)) for p in permutations(b)]
到此這篇關(guān)于關(guān)于Python兩個列表進行全組合操作的三種方式的文章就介紹到這了,更多相關(guān)Python列表全組合內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
用十張圖詳解TensorFlow數(shù)據(jù)讀取機制(附代碼)
這篇文章主要介紹了用十張圖詳解TensorFlow數(shù)據(jù)讀取機制(附代碼),小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2018-02-02python fabric實現(xiàn)遠程操作和部署示例
這篇文章主要介紹了python使用fabric實現(xiàn)遠程操作和部署示例,需要的朋友可以參考下2014-03-03