ndarray按行/列求非零元素的均值

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版权协议,转载请附上原文出处链接和本声明。