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数组,我希望list它包含50个一维数组,每个都是形状(10000,).

Answer 1

如何使用list:

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