python - 在Python中迭代单个值和可迭代的混合

Question

我正在用 Python 编写一个 for 循环，我想迭代单个对象和对象的扁平列表(或元组)的混合。

例如:

a = 'one'
b = 'two'
c = [4, 5]
d = (10, 20, 30)

我想在 for 循环中迭代所有这些。我认为这样的语法会很优雅:

for obj in what_goes_here(a, b, *c, *d):
  # do something with obj

我在itertools中查找了what_goes_here，但我没有看到任何东西，但我觉得我一定错过了一些明显的东西!

我发现的最接近的是 chain，但我想知道是否存在任何东西可以使我的示例保持不变(仅替换 what_goes_here)。

Answer 1

您可以执行此操作，但必须使用 Python 3.5 或更高版本来扩展解包语法。将所有参数放入一个容器(如元组)中，然后将该容器发送到itertools.chain。

>>> import itertools
>>> a = 'one'
>>> b = 'two'
>>> c = [4, 5]
>>> d = (10, 20, 30)
>>> list(itertools.chain((a, b, *c, *d)))
['one', 'two', 4, 5, 10, 20, 30]
>>> list(itertools.chain((a, *c, b, *d)))
['one', 4, 5, 'two', 10, 20, 30]
>>> list(itertools.chain((*a, *c, b, *d)))
['o', 'n', 'e', 4, 5, 'two', 10, 20, 30]