python - 将二维 Numpy 数组转换为一维列列表

Question

将二维 numpy 数组转换为一维列列表的最佳方法是什么？

例如，对于一个数组:

array([[ 0,  5, 10],
       [ 1,  6, 11],
       [ 2,  7, 12],
       [ 3,  8, 13],
       [ 4,  9, 14]])

我想得到:

[array([0, 1, 2, 3, 4]), array([5, 6, 7, 8, 9]), array([10, 11, 12, 13, 14])]

这个有效:

[a[:, i] for i in range(a.shape[1])]

但我想知道使用纯 Numpy 函数是否有更好的解决方案？

Answer 1

我想不出你需要什么理由

[array([0, 1, 2, 3, 4]), array([5, 6, 7, 8, 9]), array([10, 11, 12, 13, 14])]

代替

array([[0, 1, 2, 3, 4], [5, 6, 7, 8, 9], [10, 11, 12, 13, 14]])

你可以简单地用 a.T

^{如果你真的需要一个列表，那么你可以使用list(a.T)}