[python] 列表推導式和生成器表達式

列表推導式  list comprehension 

生成器表達式  generator expression

注：除方括號的變化之外，生成器表達式還不返回列表。生成器表達式返回的是生成器對象，可用作for循環中的迭代器。

例 把一個字符串變成Unicode碼位列表

>>> symbols = '[email protected]#$%^&*abc'
>>> codes = [ord(symbol) for symbol in symbols]
>>> codes
[33, 64, 35, 36, 37, 94, 38, 42, 97, 98, 99]
>>>

使用原則：只用列表推導來創建新的列表，並且盡量保持簡短（超過兩行則考慮用for循環重寫）

說明：Python會忽略代碼裡[]、{} 和() 中的換行，因此如果你的代碼裡有多行的列表、列表推導、生成器表達式、字典這一類的，可以省略不太好看的續行符\。

------------------------------------------------------

例 使用列表推導式計算笛卡兒積

>>> colors = ['black', 'white']
>>> sizes = ['S', 'M', 'L']
>>> tshirts = [(color, size) for color in colors for size in sizes]
>>> tshirts
[('black', 'S'), ('black', 'M'), ('black', 'L'), ('white', 'S'), ('white', 'M'), ('white', 'L')]
>>>

以上  for color in colors for size in sizes 先已顏色排列，再以尺碼排列，相當於如下for循環嵌套

>>> for color in colors:
... for size in sizes:
... print((color, size))
...
('black', 'S')
('black', 'M')
('black', 'L')
('white', 'S')
('white', 'M')
('white', 'L')
>>>

------------------------------------------------------

例 使用生成器表達式初始化元組

>>> symbols = '[email protected]#$%^&*abc'
>>>
>>> tuple(ord(symbol) for symbol in symbols)
(33, 64, 35, 36, 37, 94, 38, 42, 97, 98, 99)
>>>
>>> list(ord(symbol) for symbol in symbols)
[33, 64, 35, 36, 37, 94, 38, 42, 97, 98, 99]
>>>
>>> set(ord(symbol) for symbol in symbols)
{64, 33, 97, 35, 36, 37, 38, 98, 99, 42, 94}
>>>

------------------------------------------------------

例 使用生成器表達式計算笛卡兒積

>>> colors = ['black', 'white']
>>> sizes = ['S', 'M', 'L']
>>> ('%s %s'%(c, s) for c in colors for s in sizes)
>>> for tshirt in ('%s %s'%(c, s) for c in colors for s in sizes):
... print(tshirt)
...
black S
black M
black L
white S
white M
white L
>>>

參考書籍：《流暢的Python》2.2 列表推導式和生成器表達式