ValueError：在创建n维数组时设置带有序列的数组元素

Question

我想创建一个n维的numpy数组。以下是我的代码

import numpy as np
random_weights = np.empty(3)
random_weights[0] = np.array([0,1,2])
random_weights[1] = np.array([3,4,5])

上面的代码给出了ValueError: setting an array element with a sequence.错误。我正在尝试创建多维数组。这个问题的原因是什么？

Answer 1

如果你想要一个二维数组，你需要定义它。

import numpy as np

random_weights = np.empty((2, 3))  # 2 rows, 3 columns
random_weights[0] = np.array([0,1,2])
random_weights[1] = np.array([3,4,5])