在多个列表上并行进行Pythonic迭代

小编典典

python

我想知道一种优雅的Pythonic方法，可以在Python 3 中并行
地遍历列表列表（或列表的字典）。列表数量要到运行时才知道，所以我相信我不能简单地将它们作为参数提供该zip()功能。

例如，给定以下数据结构：

var = [['x1' ,'x2' ,'x3'], ['y1', 'y2', 'y3'], ['z1', 'z2', 'z3'], …]

我希望能够在每次迭代中访问以下值：

x1,y1,z1其次是x2,y2,z2依次x3,y3,z3类推。

大概我可以直接使用列表索引或使用itertools.chain实现此目的，但是我正在寻找使用生成器或其他方法的更优雅的方法。

阅读 199

2021-01-20

共1个答案

小编典典

zip(*var) 会自动打开您的列表清单。

因此，例如：

var = [['x1' ,'x2' ,'x3'], ['y1', 'y2', 'y3'], ['z1', 'z2', 'z3'], ['w1', 'w2', 'w3']]

for ltrs in zip(*var):
    print(", ".join(ltrs))

结果是

x1, y1, z1, w1
x2, y2, z2, w2
x3, y3, z3, w3

编辑： 根据下面的评论，他想使用字典中的项目，

var = {
    'id_172': ['x1', 'x2', 'x3'],
    'id_182': ['y1', 'y2', 'y3'],
    'id_197': ['z1', 'z2', 'z3']
}

我假设我们正在按键顺序使用值：

keys = sorted(var.keys())
for ltrs in zip(*(var[k] for k in keys)):
    print(", ".join(ltrs))

这使

x1, y1, z1
x2, y2, z2
x3, y3, z3

警告： 请注意，这会按照字典顺序（即字符串字母顺序）对键进行排序，例如，“ id_93” 在 “
id_101”之后。如果您的标签需要按数字顺序排序，则需要使用自定义键功能，例如

keys = sorted(var.keys(), key=lambda k: int(k[3:]))

2021-01-20