数组的平均数与numpy没有考虑零值

Question

我在努力工作，我有一个像相同的大小和形状的数组数： a= [153 186 0 258] b=[156 136 156 0] c=[193 150 950 757] 我想有平均数组，但我想该程序忽略计算中的零值。所以，这个例子的结果数组将是：d=[167.333 157.333 553 507.5] 这是这个计算的结果：d=[(153+156+193)/3 (186+136+150)/3 (156+950)/2 (258+757)/2]。有可能这样做吗？

Answer 1

>>> import numpy as np 
>>> a = np.array([153, 186, 0, 258]) 
>>> b = np.array([156, 136, 156, 0]) 
>>> c = np.array([193, 150, 950, 757]) 
>>> [np.mean([x for x in s if x]) for s in np.c_[a, b, c]] 
[167.33333333333334, 157.33333333333334, 553.0, 507.5] 

或者，也许一个更好的选择：

>>> A = np.vstack([a,b,c]) 
>>> np.average(A, axis=0, weights=A.astype(bool)) 
array([ 167.33333333, 157.33333333, 553.  , 507.5  ])