如何计算Numpy

Question

此曲线来自numpy数组

我计算曲线下面积（0到整个曲线之间的

）

np.trapz(y)

如何分别计算两个区域的曲线下面积：

• 在0和曲线之间，如果y <1
如果y> 1 ，则
• 在1.0和曲线之间

Answer 1

您可以预先选择要整合的范围，然后传递x＆amp; y向量np.trapz

示例：

xs = xs = np.arange(0, np.pi, np.pi/100000)
ys = np.sin(xs)

np.trapz(ys, xs)
# 1.9999999998355067 ~ 2

np.trapz(ys[xs<np.pi/2], xs[xs<np.pi/2])
# 0.999999999917753 ~ 1

cond = (xs>=np.pi/4) & (xs<3*np.pi/4)
np.trapz(ys[cond], xs[cond])
# 1.4141913474931518 ~ sqrt(2)

在您的特定情况下：

假设函数是这样的：

xs = np.arange(0,1,0.000001)
ys = 0.6 + xs**2

  如果y <1

，则在0和曲线之间

np.trapz(ys[ys<1], xs[ys<1])
# 0.46380019145797086

  如果y> 1，则

在1.0和曲线之间

在此处，计算y>1曲线下的面积，然后减去y=1处垂直线下的面积

i1 = np.trapz(ys[ys>1], xs[ys>1])
ys2 = np.ones(np.sum(ys>1))
i2 = np.trapz(ys2, xs[ys>1])
i1 - i2
# 0.10198754187656967

或者，作为单行：

np.trapz(ys[ys>1]-1, xs[ys>1])
# 0.10198754187656967

Answer 2

你只需将符合条件的数组部分传递给trapz：

np.trapz([i for i in y if i < 1])

并将曲线降低一个

np.trapz([i - 1 for i in y if i > 1])