Python:将多维numpy数组转换为数组列表

Question

假设我有2Darray，为简单起见，a=np.array([[1,2],[3,4]])。我想把它转换成数组的list，这样的结果是：

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

我发现有np.ndarray.tolist()函数，但它将N-D数组转换为嵌套的list。我本可以在for循环中完成（使用append方法），但它并不高效/优雅。你知道吗

在我的实际例子中，我使用的是大约10000 x 50个元素的2D数组，我希望list包含50个一维数组，每个数组的形状(10000,)。你知道吗

Answer 1

为什么不按如下方式使用列表理解而不使用任何append：

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

输出

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

Answer 2

如何使用list：

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