二维 Numpy 数组到 Numpy Arrays 列表

Question

我有一个 1000 x 1000 维度的 2D numpy 数组。当我查看类型时，我得到

>>>print(type(arr)) 
<class 'numpy.ndarray'>

>>>print(type(arr[0]))
<class 'numpy.ndarray'>

我想将此二维数组转换为 numpy arrays 的列表，并且我需要快速进行此转换，以便它适用于更大规模的 arrays。 我想转换它以便得到这些结果

>>>print(type(arr)) 
<class 'list'>

>>>print(type(arr[0]))
<class 'numpy.ndarray'>

谁能帮我解决这个问题？ 谢谢你。

Answer 1

只需使用list(arr) ：

>>> arr = np.arange(12).reshape(3, 4)
>>> arr
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> list(arr)
[array([0, 1, 2, 3]), array([4, 5, 6, 7]), array([ 8,  9, 10, 11])]

>>> print(type(arr)) 
<class 'numpy.ndarray'>

>>> print(type(arr[0]))
<class 'numpy.ndarray'>
>>> arr = list(arr)

>>> print(type(arr)) 
<class 'list'>

>>> print(type(arr[0]))
<class 'numpy.ndarray'>