1. 程式人生 > >Python之切片、迭代和列表生成式

Python之切片、迭代和列表生成式

切片

  • L=list(range(5)) 取相應範圍內的元素:L[m:n]表示從索引m開始取,直到索引n為止,但不包括索引n,如果第一個索引是0,還可以省略: L[0:3]/L[:3]——[1,2,3]
  • 還支援倒數切片: L[-2:]——[4,5] L[-2:-1]——[4]
  • 間隔取數: L[:10:2]——前10個數,每兩個取一個 L[::5]——所有數,每5個取一個
  • 複製list: L[:]——[1,2,3,4,5]
  • 同理,tuple和字串也可用切片操作

迭代

如果給定一個list或tuple,我們可以通過for迴圈來遍歷這個list或tuple,這種遍歷我們稱為迭代(Iteration)。

d =
{'a': 1, 'b': 2, 'c': 3} for key in d: print(key)#(預設情況下,dict迭代的是key。)
  • 如果要迭代value,可以用for value in d.values(),如果要同時迭代key和value,可以用for k, v in d.items()。
  • 字串也可用作for迴圈:for ch in ‘ABC’
  • 判斷一個物件是可迭代物件,是通過collections模組的Iterable型別判斷:
from collections import Iterable
isinstance('abc', Iterable)
  • Python內建的enumerate函式可以把一個list變成索引-元素對,這樣就可以在for迴圈中同時迭代索引和元素本身:
for i, value in enumerate(['A', 'B', 'C']):
print(i, value)
output:
0 A    
1 B     
2 C
  • for迴圈裡,同時引用了兩個變數
for x, y in [(1, 1), (2, 4), (3, 9)]:
print(x, y)
output:
1 1	
2 4	
3 9

列表生成式

[x * x for x in range(1, 11)]
  • 加if判斷:
[x * x for x in range(1, 11) if x % 2 == 0]
  • 可以使用兩層迴圈,可以生成全排列:
[m + n for
m in 'ABC' for n in 'XYZ'] ['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
  • 同時使用兩個甚至多個變數,比如dict的items()可以同時迭代key和value:
 d = {'x': 'A', 'y': 'B', 'z': 'C' }
 for k, v in d.items():
 print(k, '=', v)
output:
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']
  • 把一個list中所有的字串變成小寫:
L = ['Hello', 'World', 'IBM', 'Apple']
[s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']

迭代器

凡是可作用於for迴圈的物件都是Iterable型別; 凡是可作用於next()函式的物件都是Iterator型別,它們表示一個惰性計算的序列; 集合資料型別如list、dict、str等是Iterable但不是Iterator,不過可以通過iter()函式獲得一個Iterator物件。