Question

我想知道一种优雅的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来实现这一点，但我正在寻找使用生成器或其他方式更优雅的方法。

Answer 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

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

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

Answer 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

并行地对多个列表进行Pythonic迭代

2 个答案: