將二維 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]])

您可以通過aT輕松獲得

^{如果你真的需要一個列表，那么你可以使用list(aT)}

Answer 2

這是一種方法：

In [20]: list(a.T)
Out[20]: [array([0, 1, 2, 3, 4]), array([5, 6, 7, 8, 9]), array([10, 11, 12, 13, 14])]

這是另一個：

In [40]: [*map(np.ravel, np.hsplit(a, 3))]
Out[40]: [array([0, 1, 2, 3, 4]), array([5, 6, 7, 8, 9]), array([10, 11, 12, 13, 14])]

Answer 3

只需調用列表函數：

list(a.T)

Answer 4

嗯，這是另一種方式。 （我假設 x 是你原來的二維數組。）

[row for row in x.T]

但是，仍然使用 Python 列表生成器表達式。