python - 压缩两个列表列表

Question

需要帮助，因为我在压缩两个列表列表(矩阵)时迷路了。

格式完全相同的矩阵，我希望它们在相同位置的每个元素的元组对中压缩。

例如，

m1 = [['A', 'B', 'C'],
      ['D', 'E'],
      ['F', 'G']]

m2 = [['s1', 's2', 's3'],
      ['s4', 's5'],
      ['s1', 's3']]

我希望得到的是，格式相同:

z = [[('A', 's1'), ('B', 's2'), ('C', 's3')],
     [('D', 's4'), ('E', 's5')],
     [('F', 's1'), ('G', 's3')]]

我可以编写一个函数来执行此操作，但我正在寻找一种在 Python 中执行此操作的优雅方法。

Answer 1

zip() 和 zip() 再次:

[zip(*paired) for paired in zip(m1, m2)]

zip() function将输入序列的每个元素配对； m1[0] 与 m2[0]，m1[1] 与 m2[1]，等等。 ，然后对于这些对中的每一对，您再次将元素配对(m1[0][0] 与 m2[0][0]，m1[ 0][1] 与 m2[0][1] 等)。

如果这是 Python 3，则必须将其中一个包装在 list() 调用中:

[list(zip(*paired)) for paired in zip(m1, m2)]

演示:

>>> m1 = [['A', 'B', 'C'],
...       ['D', 'E'],
...       ['F', 'G']]
>>> m2 = [['s1', 's2', 's3'],
...       ['s4', 's5'],
...       ['s1', 's3']]
>>> [zip(*paired) for paired in zip(m1, m2)]
[[('A', 's1'), ('B', 's2'), ('C', 's3')], [('D', 's4'), ('E', 's5')], [('F', 's1'), ('G', 's3')]]