输入:numpy的array
输出:一个一维的平均值array
import numpy as np
def non_zero_mean(np_arr):
exist = (np_arr != 0)
num = np_arr.sum(axis=1)
den = exist.sum(axis=1)
return num/den
如果要求按行的非零元素的平均值,把所有的 axis=1改成axis=0
输入:numpy的array
输出:一个一维的平均值array
import numpy as np
def non_zero_mean(np_arr):
exist = (np_arr != 0)
num = np_arr.sum(axis=1)
den = exist.sum(axis=1)
return num/den
如果要求按行的非零元素的平均值,把所有的 axis=1改成axis=0