是否有任何快捷方式可以重新组合嵌套列表,例如
[[a0,b0,c0],[a1,b1,c1],[a2,b2,b2]]
给这样的东西
[[a0,a1,a2],[b0,b1,b2],[c0,c1,c2]]
是的,使用zip()
:
transposed = zip(*matrix)
*matrix
将所有嵌套列表matrix
作为单独的参数应用到zip()
函数中,就像您键入zip(matrix[0], matrix[1], matrix[2])
; zip()
接受输入序列并每列输出新序列。
演示:
>>> matrix = [['a0', 'b0', 'c0'], ['a1', 'b1', 'c1'], ['a2', 'b2', 'b2']]
>>> zip(*matrix)
[('a0', 'a1', 'a2'), ('b0', 'b1', 'b2'), ('c0', 'c1', 'b2')]
输出使用嵌套元组;如果您需要嵌套列表,请在事后转换它们:
map(list, zip(*matrix))
或者
[list(t) for t in zip(*matrix)]