numpy修改ndarray对角线

Question

numpy中有什么方法可以获取对角线数组的引用？我希望将阵列对角线除以某个因子，谢谢

Answer 1

如果X是您的数组并且c是因素，

X[np.diag_indices_from(X)] /= c

参见diag_indices_fromNumpy手册。