Come posso convertire un semplice elenco di elenchi in una matrice numpy? Le righe sono singole liste secondarie e ogni riga contiene gli elementi nella lista secondaria.
Come posso convertire un semplice elenco di elenchi in una matrice numpy? Le righe sono singole liste secondarie e ogni riga contiene gli elementi nella lista secondaria.
Risposte:
Se il tuo elenco di elenchi contiene elenchi con un numero variabile di elementi, la risposta di Ignacio Vazquez-Abrams non funzionerà. Invece ci sono almeno 3 opzioni:
1) Crea una matrice di array:
x=[[1,2],[1,2,3],[1]]
y=numpy.array([numpy.array(xi) for xi in x])
type(y)
>>><type 'numpy.ndarray'>
type(y[0])
>>><type 'numpy.ndarray'>
2) Crea una serie di elenchi:
x=[[1,2],[1,2,3],[1]]
y=numpy.array(x)
type(y)
>>><type 'numpy.ndarray'>
type(y[0])
>>><type 'list'>
3) Per prima cosa fai in modo che le liste siano uguali in lunghezza:
x=[[1,2],[1,2,3],[1]]
length = max(map(len, x))
y=numpy.array([xi+[None]*(length-len(xi)) for xi in x])
y
>>>array([[1, 2, None],
>>> [1, 2, 3],
>>> [1, None, None]], dtype=object)
>>> numpy.array([[1, 2], [3, 4]])
array([[1, 2], [3, 4]])
Poiché questa è la ricerca principale su Google per convertire un elenco di elenchi in un array Numpy, offrirò quanto segue nonostante la domanda abbia 4 anni:
>>> x = [[1, 2], [1, 2, 3], [1]]
>>> y = numpy.hstack(x)
>>> print(y)
[1 2 1 2 3 1]
Quando ho pensato di farlo in questo modo, ero abbastanza soddisfatto di me stesso perché è davvero semplice. Tuttavia, dopo averlo cronometrato con un elenco più ampio di elenchi, in realtà è più veloce farlo:
>>> y = numpy.concatenate([numpy.array(i) for i in x])
>>> print(y)
[1 2 1 2 3 1]
Nota che la risposta n. 1 di @ Bastiaan non crea un unico elenco continuo, quindi ho aggiunto il concatenate
.
Comunque ... preferisco l' hstack
approccio per l'uso elegante di Numpy.
Ancora una volta, dopo aver cercato il problema di convertire elenchi nidificati con livelli N in un array N-dimensionale non ho trovato nulla, quindi ecco come aggirarlo:
import numpy as np
new_array=np.array([[[coord for coord in xk] for xk in xj] for xj in xi], ndmin=3) #this case for N=3
[...[...[...]]]
parte. Devi solo chiamare np.array
, con ndmin=number-of-list-layers
. (anche se nel mio caso avevo bisogno ndmin=number-of-list-layers-minus-1
per qualche motivo, altrimenti ho creato un ulteriore livello - è necessario indagare)
np.array
semplicemente "il wrapping" di questi elenchi più profondi piuttosto che convertirli in array intorpiditi.
Avevo un elenco di elenchi di uguale lunghezza. Anche allora Ignacio Vazquez-Abrams
la risposta non ha funzionato per me. Ho un array numpy 1-D i cui elementi sono liste. Se hai riscontrato lo stesso problema, puoi utilizzare il metodo seguente
Uso numpy.vstack
import numpy as np
np_array = np.empty((0,4), dtype='float')
for i in range(10)
row_data = ... # get row_data as list
np_array = np.vstack((np_array, np.array(row_data)))