Преобразование массива NumPy в структуру списка Python?

Question

Как преобразовать массив NumPy в список Python (например, [[1,2,3],[4,5,6]]) и сделать это достаточно быстро?

Answer 1

Использование tolist():

import numpy as np 
>>> np.array([[1,2,3],[4,5,6]]).tolist() 
[[1, 2, 3], [4, 5, 6]] 

Обратите внимание, что это преобразует значения из любого типа NumPy они могут иметь (например, np.int32 или np.float32) до «ближайшего совместимого типа Python» (в списке) , Если вы хотите сохранить типы данных numpy, вместо этого вы можете вызвать list() в своем массиве, и в итоге вы получите список numpy scalars. (Благодаря Mr_and_Mrs_D для указания, что в комментарии.)

Answer 2

Numpy массивы имеют tolist method:

In [1]: arr=np.array([[1,2,3],[4,5,6]]) 

In [2]: arr.tolist() 
Out[2]: [[1, 2, 3], [4, 5, 6]] 

Answer 3

Метод NumPy .tolist производит вложенные массивы, если NumPy форма массива 2D.

Если требуются плоские списки, работает нижеописанный метод.

import numpy as np 
from itertools import chain 

a = [1,2,3,4,5,6,7,8,9] 
print type(a), len(a), a 
npa = np.asarray(a) 
print type(npa), npa.shape, "\n", npa 
npa = npa.reshape((3, 3)) 
print type(npa), npa.shape, "\n", npa 
a = list(chain.from_iterable(npa)) 
print type(a), len(a), a`