如何使用numpy和python在主對角線上方構造對角線？

Question

我想生成一個數組A定義為

A = [[1, 2, 0], [3, 1, 2], [3, 3, 1]]

我試圖將其構造為

A = np.diag(np.ones(3), k=0)

如何在主對角線上方構造對角線？ 如果我提到k = 0，它將返回主對角線上方的對角線值

Answer 1

您可以使用.fill_diagonal和一些2 .fill_diagonal 片來填充不是主要的對角線。

嘗試這個：

import numpy as np

A = np.diag(np.ones(3), k=0)
np.fill_diagonal(A[1:, :], np.ones(3) * 3)
np.fill_diagonal(A[:, 1:], np.ones(3) * 2)
print(A)