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])]