問(wèn)題

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

解決方案

itertools.chain()方法可以用來(lái)簡(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 i in chain(a, b):
    print(i, end = ' ')

1 2 3 4 x y z 

使用 chain() 的一個(gè)常見(jiàn)場(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

討論

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() 不需要這一步，如果輸入序列非常大的時(shí)候會(huì)很省內(nèi)存，并且當(dāng)可迭代對(duì)象類型不一樣的時(shí)候 chain()同樣可以很好的工作。