不同集合上元素的迭代-itertools.chain()的使用
問題
你想在多個(gè)對象執(zhí)行相同的操作，但是這些對象在不同的容器中，你希望代碼在不失可讀性的情況下避免寫重復(fù)的循環(huán)。
解決方案
itertools.chain() 方法可以用來簡化這個(gè)任務(wù)。 它接受一個(gè)可迭代對象列表作為輸入，并返回一個(gè)迭代器，有效的屏蔽掉在多個(gè)容器中迭代細(xì)節(jié)。 為了演示清楚，考慮下面這個(gè)例子：
>>> from itertools import chain
>>> a = [1, 2, 3, 4]
>>> b = ['x', 'y', 'z']
>>> for x in chain(a, b):
... print(x)
...
1
2
3
4
x
y
z
>>>
使用 chain() 的一個(gè)常見場景是當(dāng)你想對不同的集合中所有元素執(zhí)行某些操作的時(shí)候。比如：
# Various working sets of items
active_items = set()
inactive_items = set()
# Iterate over all items
for item in chain(active_items, inactive_items):
# Process item
這種解決方案要比像下面這樣使用兩個(gè)單獨(dú)的循環(huán)更加優(yōu)雅，
for item in active_items:
# Process item
...
for item in inactive_items:
# Process item
...
討論
itertools.chain() 接受一個(gè)或多個(gè)可迭代對象作為輸入?yún)?shù)。 然后創(chuàng)建一個(gè)迭代器，依次連續(xù)的返回每個(gè)可迭代對象中的元素。 這種方式要比先將序列合并再迭代要高效的多。比如：
# Inefficent
for x in a + b:
...
# Better
for x in chain(a, b):
...
第一種方案中， a + b 操作會創(chuàng)建一個(gè)全新的序列并要求a和b的類型一致。 chian() 不會有這一步，所以如果輸入序列非常大的時(shí)候會很省內(nèi)存。 并且當(dāng)可迭代對象類型不一樣的時(shí)候 chain() 同樣可以很好的工作。
