Lista list w tablicy numpy

Question

Jak przekonwertować prostą listę list na tablicę numpy? Wiersze są pojedynczymi podlistami, a każdy wiersz zawiera elementy na liście podrzędnej.

Answer 1

>>> numpy.array([[1, 2], [3, 4]]) 
array([[1, 2], [3, 4]]) 

Answer 2

Jest to tak proste, jak:

>>> lists = [[1, 2], [3, 4]] 
>>> np.array(lists) 
array([[1, 2], 
     [3, 4]]) 

Answer 3

Jeśli lista list ma list z różnej liczby elementów niż odpowiedzi Ignacio Vazquez-Abrams nie będzie działać. Zamiast tego są 3 opcje:

1) dokonać tablicy tablic:

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) Sprawdź tablicę list:

x=[[1,2],[1,2,3],[1]] 
y=numpy.array(x) 
type(y) 
>>><type 'numpy.ndarray'> 
type(y[0]) 
>>><type 'list'> 

3) Najpierw należy wykazy równej długości:

x=[[1,2],[1,2,3],[1]] 
length = len(sorted(x,key=len, reverse=True)[0]) 
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) 

Answer 4

Ponieważ jest to najlepsze wyszukiwanie w Google do konwersji listy list do tablicy Numpy, będę oferować następujące pomimo kwestii będącej 4 lat:

>>> x = [[1, 2], [1, 2, 3], [1]] 
>>> y = numpy.hstack(x) 
>>> print(y) 
[1 2 1 2 3 1] 

Kiedy pierwsza myśl to zrobić w ten sposób, byłem dość zadowolony z siebie, bo to baaaardzo proste. Jednak po taktowania go z większej listy list, to jest rzeczywiście szybciej to zrobić:

>>> y = numpy.concatenate([numpy.array(i) for i in x]) 
>>> print(y) 
[1 2 1 2 3 1] 

Zauważ, że @ Bastiaan za Odpowiedź # 1 nie robi jedną listę ciągły, a więc dodałem concatenate.

W każdym razie ... Preferuję podejście hstack dla eleganckiego korzystania z Numpy.

Answer 5

Ponownie, po wyszukaniu problemu konwersja zagnieżdżonych list z poziomu N do tablicy n-wymiarowej nic nie znalazłem, więc oto mój sposób wokół niego:

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