我创建了以下X
num_locations = 2
X= [ ]
for n in range(num_locations):
X.append([0 for j in range(num_locations)])
现在,我想将n!= m的情况下的这些X [n] [m]值相加。这样结果应该像
X[0][1]+X[1][0]
有没有办法用总和公式来做到这一点?
X[n][m] for n in range(num_locations)for m in range(num_locations))
这实际上是将2D数组中非对角元素的总和。使用Numpy的一种选择可能只是np.trace
从整个数组的总和中减去主对角线()的总和。
num_locations = 2
X= [[1,2],[2,1]]
import numpy as np
s = np.sum(X) - np.trace(X)
print(s)
输出:
4
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句