read - python list of numpy arrays




Convertendo o array NumPy na estrutura da Lista Python? (3)

Como converto uma matriz NumPy em uma lista Python (por exemplo, [[1,2,3],[4,5,6]] ) e faço isso razoavelmente rápido?


O método numpy .tolist produz matrizes aninhadas se a forma do array numpy for 2D.

Se listas planas são desejadas, o método abaixo funciona.

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`

Use tolist() :

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

Observe que isso converte os valores de qualquer tipo numpy que eles possam ter (por exemplo, np.int32 ou np.float32) para o "tipo compatível com Python mais próximo" (em uma lista). Se você quiser preservar os tipos de dados numpy, você poderia chamar list () em sua matriz, e você terminará com uma lista de escalões numpy . (Obrigado ao Mr_and_Mrs_D por apontar isso em um comentário.)


tolist() funciona bem mesmo se encontrou um array aninhado, digamos um pandas DataFrame ;

my_list = [0,1,2,3,4,5,4,3,2,1,0]
my_dt = pd.DataFrame(my_list)
new_list = [i[0] for i in my_dt.values.tolist()]

print(type(my_list),type(my_dt),type(new_list))






numpy