目錄

列表生成式

列表生成式案例1

列表生成式案例2

列表生成式案例3

列表生成式案例4

列表生成式案例5

迭代器（Iterator）

可迭代對象：Iterable

迭代器：Iterator

iter()函數

小結

列表生成式

列表生成式即List Comprehensions：

是Python內置的非常簡單卻強大的可以用來創建list的生成式。

列表生成式案例1

要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]：

可以用list(range(1, 11))：

 list(range(1, 11))

結果為：[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

但如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎麼做？

方法一是循環：

 L = []
for x in range(1, 11):
  L.append(x * x)

結果為：

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

但是循環太繁瑣，

而列表生成式則可以用一行語句代替循環生成上面的list：

[x * x for x in range(1, 11)]

結果為：[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

寫列表生成式時，把要生成的元素x * x放到前面，後面跟for循環，就可以把list創建出來，簡單、方便。

列表生成式案例2

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']

列表生成式案例3

運用列表生成式，可以寫出非常簡潔的代碼。

例如，列出當前目錄下的所有文件和目錄名，可以通過一行代碼實現：

注意：os.listdir可以列出文件和目錄

 import os # 導入os模塊，模塊的概念後面講到
[d for d in os.listdir('.')]

列表生成式案例4

for循環其實可以同時使用兩個甚至多個變量，比如dict的items()可以同時迭代key和value：

d = {'x': 'A', 'y': 'B', 'z': 'C' }
for k, v in d.items():
   print(k, '=', v)

結果為：y = B    x = A     z = C

因此，列表生成式也可以使用兩個變量來生成list：

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

結果為：['y=B', 'x=A', 'z=C']

列表生成式案例5

把一個list中所有的字符串變成小寫：

L = ['Hello', 'World', 'IBM', 'Apple']
[s.lower() for s in L]

結果為：['hello', 'world', 'ibm', 'apple']

迭代器（Iterator）

可以直接作用於for循環的數據類型有以下幾種：

一類是集合數據類型，如list、tuple、dict、set、str等；一類是generator，包括生成器和帶yield的generator function。

可迭代對象：Iterable

這些可以直接作用於for循環的對象統稱為可迭代對象：Iterable。

可以使用isinstance()判斷一個對象是否是Iterable對象：

而生成器不但可以作用於for循環，還可以被next()函數不斷調用並返回下一個值，直到最後拋出StopIteration錯誤表示無法繼續返回下一個值了。

迭代器：Iterator

可以被next()函數調用並不斷返回下一個值的對象稱為迭代器：Iterator。

可以使用isinstance()判斷一個對象是否是Iterator對象，list、dict、str雖然是Iterable，卻不是Iterator。

iter()函數

把list、dict、str等Iterable變成Iterator可以使用iter()函數：

 isinstance(iter('abc'), Iterator)

結果為：True

Iterator甚至可以表示一個無限大的數據流，例如全體自然數。而使用list是永遠不可能存儲全體自然數的。

小結

凡是可作用於for循環的對象都是Iterable類型；凡是可作用於next()函數的對象都是Iterator類型，它們表示一個惰性計算的序列；集合數據類型如list、dict、str等是Iterable，但不是Iterator，不過可以通過iter()函數獲得一個Iterator對象。