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