为什么numpy.diag无法按预期工作？

Question

我有一个如下数组：

  poles = numpy.array([[-1+1j], [-1-1j], [-2+3j], [-2-3j]])

它的形状是（4,1）。

当我使用如下所示的numpy.diag时：

LA = numpy.diag(poles)

输出为[-1.+1.j]，而我希望看到对角矩阵。有人可以解释发生了什么，应该怎么做才能看到对角矩阵？我还尝试将形状更改为（1,4），但结果没有改变。

Answer 1

您要寻找的功能是np.fill_diagonal。这将设置数组的对角线值。您必须先创建数组：

arr = np.zeros((4, 4), dtype=np.complex64))
np.fill_diagonal(arr, poles)

另一方面，

np.diagonal检索对角线中的值。 np.diagonal(arr)的输出是：

array([-1.+1.j, -1.-1.j, -2.+3.j, -2.-3.j], dtype=complex64)