4.12 不同集合上元素的迭代

問題

你想在多個(gè)對(duì)象執(zhí)行相同的操作，但是這些對(duì)象在不同的容器中，你希望代碼在不失可讀性的情況下避免寫重復(fù)的循環(huán)。

解決方案

itertools.chain() 方法可以用來簡(jiǎn)化這個(gè)任務(wù)。它接受一個(gè)可迭代對(duì)象列表作為輸入，并返回一個(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è)常見場(chǎng)景是當(dāng)你想對(duì)不同的集合中所有元素執(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è)可迭代對(duì)象最為輸入?yún)?shù)。然后創(chuàng)建一個(gè)迭代器，依次連續(xù)的返回每個(gè)可迭代對(duì)象中的元素。這種方式要比先將序列合并再迭代要高效的多。比如：

# Inefficent
for x in a + b:
    ...

# Better
for x in chain(a, b):
    ...

第一種方案中，a + b 操作會(huì)創(chuàng)建一個(gè)全新的序列并要求a和b的類型一致。chian() 不會(huì)有這一步，所以如果輸入序列非常大的時(shí)候會(huì)很省內(nèi)存。并且當(dāng)可迭代對(duì)象類型不一樣的時(shí)候 chain() 同樣可以很好的工作。