1

我有一个从文本文件中绘制数据的代码。我已经能够得到每条数据曲线下的积分（虽然我不知道其中任何一个的函数，所以我只使用了integral = s = simps(y, x)）。其中两条曲线水平重叠，我想找到它们重叠的区域。我遇到了一些麻烦，因为它们水平重叠而不是垂直重叠。我想找到的区域在这张图上的红线和蓝线之间。bx数据位于 numpy 数组中，例如by蓝线 x 和 y 以及rx红线ryx 和 y。我没有代表它们的功能，但它们在那些数组中。很难知道如何进行，因为它们水平重叠而不是垂直重叠。

我非常坚持这一点，任何帮助将不胜感激！

更新：JohanC提供的代码在一种情况下有效，但在另一种情况下无效。非工作图最初看起来像这样（我只想要那个中间重叠区域的区域），但是当它也不起作用时，我尝试限制 x 范围以希望帮助代码缩小到我想要的区域就像要计算的重叠积分（这是具有较小范围的图的来源）。我目前使用的代码是JohanC提供的代码，如下所示：

def get_overlap_integral(bx, by, rx, ry):
    fig, ax = plt.subplots()
    ax.plot(bx, by, color='b')
    ax.plot(rx, ry, color='r')

    # create a common x range
    gx = np.linspace(min(bx.min(), rx.min()), max(bx.max(), rx.max()), 1000)
    gby = np.interp(gx, bx, by)  # set bx,by onto the common x range
    gry = np.interp(gx, rx, ry)  # set rx,ry onto the common x range
    gy = np.minimum(gby, gry)  # let gx,gy be the intersection of the two curves
    ax.fill_between(gx, gy, color='g', alpha=0.3)

    area = np.trapz(gy, gx)  # calculate the green area
    ax.text(0.05, 0.95, f'Overlap: {area:.3f}', color='g', ha='left', va='top', transform=ax.transAxes)
    plt.show()
    return area

文本 bx、by、rx 和 ry 是如下值：

BX: [999.5 999.  998.5 ... 201.  200.5 200. ]
BY: [-3.786867e-05 -4.366451e-05 -4.745308e-05 ...  1.068685e-05  1.555391e-05
 -8.949840e-06]
RX: [999.5 999.  998.5 ... 201.  200.5 200. ]
RY: [-5.865443e-05 -7.808241e-05 -5.887286e-05 ... -1.556630e-06 -3.473830e-06
 -6.367470e-06]

我不确定为什么此功能不适用于一种情况，但会完美地适用于另一种情况。任何帮助将不胜感激！

4

1 回答 1

0

np.interp(new_x, old_x, old_y)可以为新的 x 范围计算曲线。在公共 x 范围上设置时，np.minimum查找两条正值曲线的公共区域。 np.trapz计算面积。

下面是一些示例代码，从创建测试数据开始：

from matplotlib import pyplot as plt
import numpy as np

# create some test data
Nb = 300
bx = np.linspace(320, 750, Nb)
by = np.random.randn(Nb).cumsum()
by -= by[0]  # start at zero
by -= np.linspace(0, 1, Nb) * by[-1]  # let end in zero
by = (by ** 2) ** .3  # positive curve starting and ending with zero
Nr = 200
rx = np.linspace(610, 1080, Nr)
ry = np.random.randn(Nr).cumsum()
ry -= ry[0]  # start at zero
ry -= np.linspace(0, 1, Nr) * ry[-1]  # let end in zero
ry = (ry ** 2) ** .3  # positive curve starting and ending with zero

fig, ax = plt.subplots()
ax.plot(bx, by, color='b')
ax.plot(rx, ry, color='r')

# create a common x range
gx = np.linspace(min(bx.min(), rx.min()), max(bx.max(), rx.max()), 1000)
gby = np.interp(gx, bx, by)  # set bx,by onto the common x range
gry = np.interp(gx, rx, ry)  # set rx,ry onto the common x range
gy = np.minimum(gby, gry)  # let gx,gy be the intersection of the two curves
ax.fill_between(gx, gy, color='g', alpha=0.3)

area = np.trapz(gy, gx)  # calculate the green area
ax.text(0.05, 0.95, f'Overlap: {area:.3f}', color='g', ha='left', va='top', transform=ax.transAxes)
plt.show()

找到具有不同 x 范围的两条曲线的公共区域

于 2021-07-12T16:32:22.817 回答