Python：將多維numpy數組轉換為數組列表

Question

假設我有2D array ，為簡單起見， a=np.array([[1,2],[3,4]]) 。 我想將其轉換為數組list ，這樣結果將是：

b=[np.array([1,2]), np.array([3,4])]

我發現有np.ndarray.tolist()函數，但是它將ND數組轉換為嵌套list 。 我本可以在for循環中完成（使用append方法），但這不是有效/優雅的方法。

在我的真實示例中，我正在使用大約10000 x 50元素的2D數組，並且我想要包含50個一維數組的list ，每個數組的形狀為(10000,) 。

Answer 1

如何使用list ：

a=np.array([[1,2],[3,4]])
b = list(a)

Answer 2

為什么不按以下方式使用列表推導而不使用任何append ：

a=np.array([[1,2],[3,4]])
b = [i for i in a]
print (b)

輸出量

[array([1, 2]), array([3, 4])]