假如我们有一个矩阵A,想要求其中的非零元素的均值,计算方式是:
直接用mean来求的话,分母是所有元素的个数,不符合我们的要求。
在matlab中,这样做:
mean(AA(AA~=0))
numpy:
import numpy as np
a = np.array([[0,1],[0,0]])
exist = (a != 0)
mean_value = a.sum()/exist.sum()
先用着这种方法吧,简洁的没找到。
参考:https://blog.csdn.net/iteapoy/article/details/85139485