python - 如何压缩单个嵌套列表的元素

Question

这个问题已经有答案了:

Expanding tuples into arguments (5 个回答)

已关闭 3 年前。

考虑嵌套列表:

d = [[1,2,3],[4,5,6]]                                                                                                                                                                                                                                                  

我想要zip此结果的元素:

 [[1,4],[2,5],[3,6]]

如何做到这一点？错误的做法是

list(zip(d))

但是这给出了:

[([1, 2, 3],), ([4, 5, 6],)]

执行 zip 的正确方法是什么？ ？

Answer 1

您需要通过解包 (*) 将单个子列表作为 zip() 的单个参数提供，如下所示:

d = [[1,2,3],[4,5,6]]          
zip(*d)  # You need this one
[(1, 4), (2, 5), (3, 6)]

这甚至适用于较长的列表，如果这是您想要的行为:

zip(*[[1,2,3],[4,5,6],[7,8,9]])
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]

如果您想要一个列表列表而不是元组列表，只需执行以下操作:

map(list, zip(*d))
[[1, 4], [2, 5], [3, 6]]