python - 如何从列表列表中创建一个 numpy 数组

Question

现在我有一个列表列表，其中包含不同组的坐标。前任。列表 = [ [ [0,3],[4,1],[2,3] ], [ [2,5],[41,12],[2,3] ] ]

我想从每组坐标中创建一个 numpy 数组，但我不知道如何转换这个列表。在我的情况下，组的数量总是在变化，所以无论如何循环我的列表并制作 len(list) 数量的 numpy 数组？

标签： pythonnumpy

Answer 1

您可以只调用np.array您的输入，例如：

data = [[[0, 3], [4, 1], [2, 3]], [[2, 5], [41, 12], [2, 3]]]
array = np.array(data)

array.shape

输出：

(2, 3, 2)

如果组的数量（我认为外部列表中的元素数量）发生变化，只要内部列表的形状没有变化，这很好。

结果的形状通常是(len(data), 3, 2)。