有没有办法让numpy矩阵存储对象？

Question

我想在 numpy 矩阵中存储一个元组，但它似乎会返回一个错误。有办法吗？

>>> import numpy
>>> y = numpy.zeros((4,4))
>>> y[1][1] = (1,1)

ValueError: setting an array element with a sequence.

谢谢

Answer 1

使用dtype=object，你可以将任何你想要的东西放入你的数组中：

>>> arr = np.zeros((4, 4), dtype=object)
>>> arr[1, 1] = (1, 1)
>>> arr
array([[0, 0, 0, 0],
       [0, (1, 1), 0, 0],
       [0, 0, 0, 0],
       [0, 0, 0, 0]], dtype=object)