Skip to main content
 首页 » 编程设计

Python 并行遍历多个列表

2025年05月04日10soundcode

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

请您参考如下方法:

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