python - 2D Numpy 数组到 Numpy 数组列表

Question

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

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

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

我想将此 2D 数组转换为 numpy 数组列表，并且我需要快速进行此转换，以便它适用于更大范围的数组。我想对其进行转换，以便获得这些结果

>>>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'>