Como converter uma lista simples de listas em uma matriz numpy? As linhas são sublistas individuais e cada linha contém os elementos na sub-lista.
Como converter uma lista simples de listas em uma matriz numpy? As linhas são sublistas individuais e cada linha contém os elementos na sub-lista.
Respostas:
Se sua lista de listas contiver listas com um número variável de elementos, a resposta de Ignacio Vazquez-Abrams não funcionará. Em vez disso, existem pelo menos três opções:
1) Faça uma matriz de matrizes:
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) Faça uma matriz de listas:
x=[[1,2],[1,2,3],[1]]
y=numpy.array(x)
type(y)
>>><type 'numpy.ndarray'>
type(y[0])
>>><type 'list'>
3) Primeiro faça as listas iguais em comprimento:
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]])
Como esta é a principal pesquisa no Google para converter uma lista de listas em um array Numpy, oferecerei o seguinte, apesar da pergunta ter 4 anos:
>>> x = [[1, 2], [1, 2, 3], [1]]
>>> y = numpy.hstack(x)
>>> print(y)
[1 2 1 2 3 1]
Quando pensei em fazê-lo dessa maneira, fiquei bastante satisfeito comigo mesmo porque é muuuuito simples. No entanto, depois de cronometrar com uma lista maior de listas, é realmente mais rápido fazer isso:
>>> y = numpy.concatenate([numpy.array(i) for i in x])
>>> print(y)
[1 2 1 2 3 1]
Note que a resposta # 1 de Bastiaan não faz uma única lista contínua, por isso adicionei o concatenate
.
Enfim ... Prefiro a hstack
abordagem pelo seu uso elegante do Numpy.
Novamente, depois de procurar o problema de converter listas aninhadas com níveis N em uma matriz N-dimensional, não encontrei nada, então aqui está o meu caminho:
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
[...[...[...]]]
peça. Você só precisa ligar np.array
, com ndmin=number-of-list-layers
. (embora no meu caso eu precisasse, ndmin=number-of-list-layers-minus-1
por algum motivo, criei uma camada extra - preciso investigar)
np.array
com que "envolvesse" essas listas mais profundas em vez de convertê-las em matrizes numpy.
Eu tinha uma lista de listas de igual comprimento. Mesmo assim Ignacio Vazquez-Abrams
, a resposta não funcionou para mim. Eu tenho uma matriz numpy 1-D cujos elementos são listas. Se você enfrentou o mesmo problema, pode usar o método abaixo
Usar 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)))