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

Question

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

我想从每组坐标中创建一个 numpy 数组，但我不知道如何转换这个列表。就我而言，组的数量总是在变化，所以有没有办法循环遍历我的列表并生成 len(list) numpy 数组的数量？

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)。