并行对多个列表进行 Pythonic 迭代答案

【问题标题】：Pythonic iteration over multiple lists in parallel并行对多个列表进行 Pythonic 迭代
【发布时间】：2014-03-21 14:42:51
【问题描述】：

我想知道在 Python 3 中并行迭代列表列表（或列表的字典）的优雅的 Pythonic 方法。列表的数量直到运行时才知道，所以我相信我不能简单地将它们作为参数提供给zip() 函数。

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

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

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

x1,y1,z1 后跟 x2,y2,z2 后跟 x3,y3,z3 等等。

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

【问题讨论】：

相关：Matrix Transpose in Python

标签： python for-loop python-3.x data-structures

【解决方案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

编辑：下面的每个 cmets，他想使用字典中的项目，

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:]))

【讨论】：

奇妙、滑稽和令人尴尬的优雅。非常感谢。
我在问题中还提到的是列表字典的情况。您知道我如何以这种方式并行访问 dict 的键和值（包含列表）吗？大概我不能将.items() 与zip() 一起使用？
@bedeabc：你能给出一个示例字典输入和输出吗？
非常感谢@hugh。类似{'id_172': ['x1', 'x2', 'x3'], 'id_182': ['y1', 'y2', 'y3'], 'id_197': ['z1', 'z2', 'z3']}
@bedeabc：两个问题：首先，dict 没有排序；你想要任何特定顺序的值吗？第二，你想对键做任何事情，还是可以忽略它们？

【解决方案2】：

如果您仔细查看输入和输出数据，您实际上是在转置一个二维数组。这可以通过 Python 的内置 zip 函数轻松完成，就像这样

var = [['x1' ,'x2' ,'x3'], ['y1', 'y2', 'y3'], ['z1', 'z2', 'z3']]
print list(zip(*var))
# [('x1', 'y1', 'z1'), ('x2', 'y2', 'z2'), ('x3', 'y3', 'z3')]

你可以像这样迭代它

for x, y, z in zip(*var):
   print x, y, z

输出

x1 y1 z1
x2 y2 z2
x3 y3 z3

【讨论】：