Преобразование списка списков в массив NumPy

Обычной задачей анализа данных является преобразование списка списков в массив NumPy для эффективные числовые операции. Этот массив можно сформировать путем присвоения каждому списку строки, при этом каждый элемент списка занимает столбец.

Вариант 1: Массив массивов

Если подсписки имеют разную длину, подходящим подходом является создание массива массивов. Это сохраняет исходную структуру списка списков, что позволяет легко извлекать определенные элементы или выполнять операции с отдельными подсписками.

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

Вариант 2: Массив списков

Альтернативный метод — создать массив списков. Этот подход поддерживает структуру списка списков, при этом каждый подсписок представлен как список в массиве.

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

Вариант 3: Единая длина списков

Если важно, чтобы подсписки имели одинаковую длину, более короткие списки можно дополнить параметром None ценности. При этом создается прямоугольный массив с одинаковыми размерами.

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