python - 使用 itertools.cycle() 循环遍历多个列表

Question

我有一个服务器列表。每个服务器上都有一个名称列表。示例:

server1 = ['a','b','c']
server2 = ['d','e','f']
server3 = ['g','h','i']

我想迭代每个服务器名称而不是每个服务器。例如，在 server1 中选择 'a' 后，移动到 'd'(不是 'b')并很快。如果我要使用 itertools.cycle()，我是否必须创建要循环的服务器列表？我的预期结果是 ['a','d','g','b','e','h','c','f','i']。你能给我一个关于如何在多个列表中循环的简单例子吗？

Answer 1

我们也可以使用itertools.chain.from_iterable()相比之下，速度更快。

import itertools

server1 = ['a','b','c']
server2 = ['d','e','f']
server3 = ['g','h','i']

print list(itertools.chain.from_iterable(zip(server1,server2,server3)))

结果:

['a', 'd', 'g', 'b', 'e', 'h', 'c', 'f', 'i']