NumPy-array converteren naar Python List-structuur?

Hoe converteer ik een NumPy-array naar een Python-lijst (bijvoorbeeld [[1,2,3],[4,5,6]]), en doe je het redelijk snel?


Antwoord 1, autoriteit 100%

Gebruik tolist():

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

Merk op dat dit de waarden converteert van welk numpy-type ze ook hebben (bijv. np.int32 of np.float32) naar het “dichtstbijzijnde compatibele Python-type” (in een lijst). Als u de numpy-gegevenstypen wilt behouden, kunt u in plaats daarvan list() in uw array aanroepen, en u krijgt een lijst met numpy scalars. (Met dank aan Mr_and_Mrs_D voor het erop wijzen in een opmerking.)


Antwoord 2, autoriteit 3%

De numpy .tolist-methode produceert geneste lijsten als de numpy-arrayvorm 2D is.

als platte lijsten gewenst zijn, werkt de onderstaande methode.

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`

Antwoord 3, autoriteit 2%

c = np.array([[1,2,3],[4,5,6]])

list(c.flatten())

Antwoord 4

tolist()Werkt prima, zelfs als je een geneste array hebt aangetroffen, zeg een panda’s 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))

Antwoord 5

een andere optie

c.ravel()
# or
c.ravel().tolist()

werkt ook.

Other episodes