python:清單生成器

今天來聊聊python的清單生成器

最簡單的:

[x for x in range(10)]
           

得到的結果是:

[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

稍微複雜一點的:

[x for x in 'abcdefg']           

結果:

['a', 'b', 'c', 'd', 'e', 'f', 'g']

沒錯,這個可以把字元串轉成單個字元的清單

msg='abcdefg'
print([x for x in msg])           

總之,這樣可以把一個可疊代對象拆開,比如:

這樣(字典):

msg={1:'abc',2:'bbc'}
print([x for x in msg.items()])           

[(1, 'abc'), (2, 'bbc')]

這樣(集合):

msg={'abc',2}
print([x for x in msg])           

[2, 'abc']

還有這樣(元組):

msg=('abc',2,3,'4')
print([x for x in msg])           

['abc', 2, 3, '4']

另外還有操作可以給生成的清單加一些條件:

msg=('abc',2,3,'4','567')
print([x for x in msg if isinstance(x,int)])           

[2, 3]

或者多重循環的嵌套:

msg1=('a','b','c')
msg2=(1,2,3)
print([x+str(y) for x in msg1 for y in msg2] )           

['a1', 'a2', 'a3', 'b1', 'b2', 'b3', 'c1', 'c2', 'c3']

那麼我們要知道從三個集合裡面取出來的元素所有的組合(注意不是排列)就可以這樣:

msg1=('a','b','c')
msg2=(1,2,3)
msg3=('%','$')
res=[(x,y,z) for x in msg1 for y in msg2 for z in msg3]
print(res)
print(len(res))           

[('a', 1, '%'), ('a', 1, '$'), ('a', 2, '%'), ('a', 2, '$'), ('a', 3, '%'), ('a', 3, '$'), ('b', 1, '%'), ('b', 1, '$'), ('b', 2, '%'), ('b', 2, '$'), ('b', 3, '%'), ('b', 3, '$'), ('c', 1, '%'), ('c', 1, '$'), ('c', 2, '%'), ('c', 2, '$'), ('c', 3, '%'), ('c', 3, '$')]

18