python - 如何在numpy中计算多个平均值？

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 数组。

标签： pythonnumpy

Answer 1

您可以确定哪些元素大于或小于0.5，然后使用它来索引y

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