本文實例講述了Python編程使用*解包和itertools.product()
求笛卡爾積的方法。分享給大家供大家參考,具體如下:
【問題】
目前有一字符串s = "['a', 'b'],['c', 'd']"
,想把它分開成為兩個列表:
1
2
|
list1 = [ 'a' , 'b' ] list2 = [ 'c' , 'd' ] |
之后使用itertools.product()
求笛卡爾積,應該寫成:
1
2
|
for i in itertools.product(list1, list2): print i |
結(jié)果為:
1
2
3
4
|
( 'a' , 'c' ) ( 'a' , 'd' ) ( 'b' , 'c' ) ( 'b' , 'd' ) |
然而使用eval(s)獲得的是一個元組。product的參數(shù)如果是元組則一定會報錯(product的參數(shù)是兩個列表,每個列表中的元素數(shù)量不定)。怎么破?
【解答】
其實只差一個*而已。*是python中一個賦值的技巧,叫做解包。相信很多人都見過def func(*args, **kwargs)
這種寫法,在函數(shù)中,*代表不定個數(shù)的參數(shù),以tuple的方式傳入,**則是以dict的方式。在使用函數(shù)的時候,也可以有類似的方法,調(diào)用func(*args)函數(shù)時,相當于把一個元組args拆開,當成參數(shù)傳進函數(shù)中。只是這樣做要小心的是,args中含有的元素數(shù)量及類型必須跟該函數(shù)定義一致,否則會報SyntaxError: invalid syntax語法錯誤。
例如,在這道題中,就可以寫成:
1
2
|
for i in itertools.product( * eval (s)): print i |
就可以出來結(jié)果了。
這個問題用到的三個技巧:
(1)itertools.product()
求笛卡爾積。itertools這個模塊中有相當多的牛逼閃閃的數(shù)學算法,比如全排列函數(shù)permutations,組合函數(shù)combinations等等,有時候想要一個數(shù)學類的函數(shù)又不想自己寫,可以在這里找找,沒準有驚喜。
(2)eval()
字符串求值。eval和exec這兩個python中的逆天函數(shù),強大到讓人不太放心其安全性。
(3)*解包。上面已經(jīng)解釋過了,其實用到的場合感覺挺有限的,有印象即可,能在無路可走的時候靈光一現(xiàn)就好,別太指望它給你的程序帶來多大好處。
希望本文所述對大家Python程序設(shè)計有所幫助。
原文鏈接:https://www.cnblogs.com/anpengapple/p/5051013.html