用于选择绘图区域的矩形并找到最大值

Question

我正在寻找有关从哪里开始编写我的最新项目的建议.

我希望能够在matplotlib图上绘制稍微透明的矩形,并返回此正方形内的值.该项目的最终目标是返回该区域中的最大值,并在以后用于进一步分析.

这是我以前没有亲眼见过的事情,也不知道从哪个项目开始.我希望用这个代码指出正确的方向.

感谢您提供任何建议.

Answer 1

要以交互方式绘制矩形,您可以使用a RectangleSelector.matplotlib页面有一个例子.

然后,您可以找出矩形范围内的哪些点,并使用相应的numpy函数找到最大值.

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.widgets  import RectangleSelector

xdata = np.linspace(0,9*np.pi, num=301)
ydata = np.sin(xdata)*np.cos(xdata*2.4)

fig, ax = plt.subplots()
line, = ax.plot(xdata, ydata)
point, = ax.plot([],[], marker="o", color="crimson")
text = ax.text(0,0,"")

def line_select_callback(eclick, erelease):
    x1, y1 = eclick.xdata, eclick.ydata
    x2, y2 = erelease.xdata, erelease.ydata

    mask= (xdata > min(x1,x2)) & (xdata < max(x1,x2)) & \
          (ydata > min(y1,y2)) & (ydata < max(y1,y2))
    xmasked = xdata[mask]
    ymasked = ydata[mask]

    if len(xmasked) > 0:
        xmax = xmasked[np.argmax(ymasked)]
        ymax = ymasked.max()
        tx = "xmax: {:.3f}, ymax {:.3f}".format(xmax,ymax)
        point.set_data([xmax],[ymax])
        text.set_text(tx)
        text.set_position((xmax,ymax))
        fig.canvas.draw_idle()


rs = RectangleSelector(ax, line_select_callback,
                       drawtype='box', useblit=False, button=[1], 
                       minspanx=5, minspany=5, spancoords='pixels', 
                       interactive=True)

plt.show()