欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

python求numpy中array按列非零元素的平均值案例

程序员文章站 2022-06-21 08:41:07
输入:numpy的array输出:一个一维的平均值arrayimport numpy as np def non_zero_mean(np_arr): exist = (np_arr != 0) nu...

输入: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

补充知识:python dataframe 统计行列中零值的个数

1、按行统计,返回为一个series:

(df == 0).astype(int).sum(axis=1)

以上这篇python求numpy中array按列非零元素的平均值案例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。