python求numpy中array按列非零元素的平均值
输入:numpy的array输出:一个一维的平均值arrayimport numpy as npdef non_zero_mean(np_arr):exist = (np_arr != 0)num = np_arr.sum(axis=1)den = exist.sum(axis=1)return num/den如果要求按行的非零元素的平均...
·
输入: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
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)