Y=[1,2,3,4,5,0,0,0,0]
np.average(Y)
Out[178]: 1.6666666666666667
sum(Y)/len(Y)
Out[179]: 1.6666666666666667
sum(Y)/len(np.nonzero(Y)[0])
Out[180]: 3.0
Y=[1,2,3,4,5,0,0,0,0]
np.average(Y)
Out[178]: 1.6666666666666667
sum(Y)/len(Y)
Out[179]: 1.6666666666666667
sum(Y)/len(np.nonzero(Y)[0])
Out[180]: 3.0