numpy更改矩阵对角线上元素的值

赋不同值

import numpy as np data = score = np.array([[80, 89, 86, 67, 79], [78, 97, 89, 67, 81], [90, 94, 78, 67, 74], [91, 91, 90, 67, 69], [76, 87, 75, 67, 86]]) row, col = np.diag_indices_from(data) data[row,col] = np.array([1, 2, 3, 4, 5]) print(data) [[ 1 89 86 67 79] [78 2 89 67 81] [90 94 3 67 74] [91 91 90 4 69] [76 87 75 67 5]]

赋相同值

import numpy as np data = score = np.array([[80, 89, 86, 67, 79], [78, 97, 89, 67, 81], [90, 94, 78, 67, 74], [91, 91, 90, 67, 69], [76, 87, 75, 67, 86]]) row, col = np.diag_indices_from(data) data[row,col] = 100 print(data) [[100 89 86 67 79] [ 78 100 89 67 81] [ 90 94 100 67 74] [ 91 91 90 100 69] [ 76 87 75 67 100]]