Convertendo uma lista de listas em um array NumPy

Uma tarefa comum na análise de dados é converter uma lista de listas em um array NumPy para operações numéricas eficientes. Este array pode ser formado atribuindo cada lista a uma linha, com cada elemento da lista ocupando uma coluna.

Opção 1: Array de Arrays

Se as sublistas têm comprimentos variados, uma abordagem adequada é criar uma matriz de matrizes. Isso preserva a estrutura original da lista de listas, facilitando a recuperação de elementos específicos ou a execução de operações em sublistas individuais.

x = [[1, 2], [1, 2, 3], [1]]
y = numpy.array([numpy.array(xi) for xi in x])

Opção 2: Matriz de listas

Um método alternativo é criar um matriz de listas. Essa abordagem mantém a estrutura da lista de listas, com cada sublista representada como uma lista dentro da matriz.

x = [[1, 2], [1, 2, 3], [1]]
y = numpy.array(x)

Opção 3: Comprimentos de lista uniformes

Se for essencial que as sublistas tenham comprimentos uniformes, é possível preencher listas mais curtas com None valores. Isso cria uma matriz retangular com dimensões consistentes.

x = [[1, 2], [1, 2, 3], [1]]
length = max(map(len, x))
y = numpy.array([xi   [None] * (length - len(xi)) for xi in x])