numpy 列表数组到二维数组

Question

>>> print(a)
 
array([list([52, 584, 78]), list([35, 140, 584])], dtype=object)

我想将上面的数据转换为带有 int 的 numpy 二维数组，但是当我输入 np.array(a) 时，结果和以前一样。 如何转换类型？

Answer 1

你想使用 np.stack

>>> import numpy as np
>>> a = ([1,2,3,4,5])
>>> b = ([2,3,4,5,6])
>>> c = ([3,4,5,6,7])

>>> np.array([a, b, c])
array([[1, 2, 3, 4, 5],
       [2, 3, 4, 5, 6],
       [3, 4, 5, 6, 7]])

>>> np.stack([a, b, c], axis=0)
array([[1, 2, 3, 4, 5],
       [2, 3, 4, 5, 6],
       [3, 4, 5, 6, 7]])