numpy Python 多维数组 求非零位置的平均数

Python numpy多维数组 求非零位置的平均数

import numpy as np

# method 1 把多维数组变成一维数组才好继续操作
b_mean = b.flatten()[np.nonzero(b.flatten())].mean()

# method 2
exist = (b != 0)
b_mean = b.sum() / exist.sum()

两种方法结果几乎一样。