我用a = np.zeros((3,3))
创建了一个全为零的3x3 np.array。
我还有另一个3x1 numpy数组b = np.array([[1],[2],[3]])
,我想为其中的对角线值设置这些值。因此,作为结果,我希望有一个矩阵,对角线上的b值为b,其他所有位置的值为零。
我尝试通过以下代码使用np.fill_diagonal()函数:
a = np.zeros((3,3))
b = np.array([[1],[2],[3]])
print(np.fill_diagonal(a,b))
结果我得到:
没有
我做错了什么,我该怎么做?