在Python中，用行中其他元素的总和替换矩阵的对角元素

Question

import pandas as pd
import numpy as np
rates=(pd.read_excel("C:\Anaconda3\RateMatrix.xlsx", sheetname="Pu239Test", skiprows=0)).as_matrix() #read the matrix values from excel spreadsheet, and converts the values to a matrix

费率为22 x 22矩阵。

我想用行中所有其他元素的总和替换费率矩阵的对角线元素。

例如，

费率。项目（0,0）=费率。项目（0,1）费率。项目（0,2）费率。项目（0,3）。。。。费率。项目（0,21）

rates.item（1,1）=rates.item（1,0）rates.item（1,2）rates.item（1,3）rates.item（1,21）

.....

费率。第（21,21）项=费率。项目（21,0）费率。第（21,2）项费率。项目（21,3）。。。。费率。项目（21,20）

我在想我怎么能做到。提前多谢了。

Answer 1

这是一种矢量化方法，它使用NumPy数组a作为输入-

In [171]: a        # Input array
Out[171]: 
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24]])

# Get row and column indices of diag elements    
In [172]: row,col = np.diag_indices_from(a)

# Assign the sum of each row except the diag elems into diag positions
In [173]: a[row,col] = a.sum(axis=1) - a[row,col]

# Updated array
In [174]: a
Out[174]: 
array([[10,  1,  2,  3,  4],
       [ 5, 29,  7,  8,  9],
       [10, 11, 48, 13, 14],
       [15, 16, 17, 67, 19],
       [20, 21, 22, 23, 86]])

让我们手动计算求和并对照对角线元素进行交叉检查-

In [175]: a[0,1] + a[0,2] + a[0,3] + a[0,4]
Out[175]: 10

In [176]: a[1,0] + a[1,2] + a[1,3] + a[1,4]
Out[176]: 29

In [177]: a[2,0] + a[2,1] + a[2,3] + a[2,4]
Out[177]: 48

In [178]: a[3,0] + a[3,1] + a[3,2] + a[3,4]
Out[178]: 67

In [179]: a[4,0] + a[4,1] + a[4,2] + a[4,3]
Out[179]: 86