def non_zero_mean(np_arr, axis=0):
"""ndarray按行/列求非零元素的均值。
axis=0按列
axis=1按行
"""
exist = (np_arr != 0)
num = np_arr.sum(axis=axis)
den = exist.sum(axis=axis)
return num/den
版权声明:本文为weixin_44695969原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。