将列表列表转换为 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])