利用到Numpy库函数 numpy.diag_indices_from
import numpy as np
#3×3的单位矩阵
a = np.eye(3)
#获取主对角线元素的索引
row, col = np.diag_indices_from(a)
#对主对角线进行赋值
a[row, col] = [
利用到Numpy库函数 numpy.diag_indices_from
import numpy as np
#3×3的单位矩阵
a = np.eye(3)
#获取主对角线元素的索引
row, col = np.diag_indices_from(a)
#对主对角线进行赋值
a[row, col] = [