python - numpy配列の下部と上部の対角値を置き換える関数はnumpyにありますか？

Question

私が使用した主対角線を置き換えるためにnp.fill_diagonal：

matrix = np.zeros((4, 4), float)
main = np.array([2,2,2,2])
np.fill_diagonal(matrix, main)

ただし、主対角線の隣にある上下の対角線も置き換える必要があります。

upper=np.array([1,1,1])
lower=np.array([7,7,7])

取得するため：

matrix=[[2 1 0 0]
        [7 2 1 0]
        [0 7 2 1]
        [0 0 7 2]]

ありがとうございました

Answer 1

スマートスライスを使用すると、np.fill_diagonalこれも実行できます。

>>> np.fill_diagonal(matrix[:-1, 1:], upper)
>>> np.fill_diagonal(matrix[1:, :-1], lower)
>>> matrix
array([[ 2.,  1.,  0.,  0.],
       [ 7.,  2.,  1.,  0.],
       [ 0.,  7.,  2.,  1.],
       [ 0.,  0.,  7.,  2.]])