1. 程式人生 > 程式設計 >python求numpy中array按列非零元素的平均值案例

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

輸入: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按列非零元素的平均值案例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。