如何计算numpy的多个平均值？

Question

In the case,

import numpy as np
import matplotlib.pyplot as plt

x=np.linspace(0,1,100)
y=x**2

result=np.average(y)
print(result)

I need to calculate two sets of average as

if y>.5 :
    result1=np.average(y=.5)
    result2.np.average(y-.5)
if y<=.5
    result1=np.average(y)

result1 has a maximum y value, and result2 a minimum. However, I cannot apply the if condition for a numpy array.

Answer 1

You can determine which elements are greater or less than 0.5 , then use that to index back into y

>>> np.average(y[y>0.5])
0.7443118049178656
>>> np.average(y[y<0.5])
0.16784001632486484