Как рассчитать кратное среднее в numy?

Question

В случае

import numpy as np
import matplotlib.pyplot as plt

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

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

Мне нужно рассчитать два набора средних как

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

result1 имеет максимальное значение y, а result2 минимальное. Однако я не могу применить условие if для массива numpy.

Answer 1

Вы можете определить, какие элементы больше или меньше 0.5, а затем использовать это для индексации обратно в y

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