列表生成式

列表生成式即List Comprehensions,是Python內置的非常簡單卻強大的可以用來創建list的生成式。

[x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

寫列表生成式時,把要生成的元素x * x放到前面,後面跟for循環,就可以把list創建出來
for循環後面還可以加上if判斷,這樣我們就可以篩選出僅偶數的平方:

[x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]

還可以使用兩層循環,可以生成全排列:

[m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

運用列表生成式,可以寫出非常簡潔的代碼。例如,列出當前目錄下的所有文件和目錄名,可以通過一行代碼實現:

import os
[ d for d in os.listdir('.')]
['.bash_history', 'bin.tar', '.bash_logout', '.bash_profile', 'intedio', 'data', 'kerneldump', 'bin', 'share', '.viminfo', 'logs', '.bashrc', '.lesshst', 'etc', 'tmp', 'nvidia-dump', '.ssh', 'boomata_full.tar', 'lib']

列表生成式也可以使用兩個變量來生成list:

d = {'x': 'A', 'y': 'B', 'z': 'C' }
[k + '=' + v for k, v in d.iteritems()]
['y=B', 'x=A', 'z=C']

最後把一個list中所有的字符串變成小寫:

L = ['Hello', 'World', 'IBM', 'Apple']
[s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章