我有一个如下数组:
poles = numpy.array([[-1+1j], [-1-1j], [-2+3j], [-2-3j]])
它的形状是(4,1)。
当我使用如下所示的numpy.diag时:
LA = numpy.diag(poles)
输出为[-1.+1.j]
,而我希望看到对角矩阵。有人可以解释发生了什么,应该怎么做才能看到对角矩阵?我还尝试将形状更改为(1,4),但结果没有改变。
您要寻找的功能是np.fill_diagonal
。这将设置数组的对角线值。您必须先创建数组:
arr = np.zeros((4, 4))
np.fill_diagonal(arr, poles)
[np.diagonal
检索对角线中的值。