1. 单条图线的修改
1. 假设我有一个画图结果,比如画出来y = 1 和 y = 3 的两条线, 但是用户不满意,用户希望是 y = 1.5 和 y = 3.1,那么就需要用户自己拖动鼠标进行修改。以下代码可以达成这个效果。
2. 但是问题是,这段代码每次只能修改一条线,意思是要么改y = 1,要么就是改y = 3。
3. 代码中用到了matplotlib的鼠标事件,可以参考plt.connect(s, func)函数
# -*- coding: utf-8 -*-
# @Time : 2023/4/23 21:16
# @Author : TuoTuo
# @Software: PyCharm
from matplotlib.backend_bases import MouseButton
import matplotlib.pyplot as plt
import numpy as np
'''当前的问题:
无法一次性修改多个,只能针对某个线修改一次!
'''
# 假设我们有2条线可以修改
# 目标就是在一定误差范围内拖动这两根线
def on_move(event):
if event.inaxes:
for y_values in [1.2,3]:
# 如果newy放在for循环前面的话
# 会报错,因为此时newy里面会有200个值
newy = []
# 设置阈值,设置是 1
if -1 < event.ydata - y_values < 1:
for i in range(0,len(t)):
newy.append(event.ydata)
#清空当前图区
ax.cla()
#画出三条曲线作为对比
ax.plot(t,s,'r',marker = '+')
ax.plot(t,s1,'g')
ax.plot(t,newy)
plt.ylim(-1,5)
plt.legend(['last one','next one','modified one'])
# 这行代码可以实时更新你的图
fig.canvas.draw_idle()
def on_press(event):
plt.connect('motion_notify_event', on_move)
def on_button_release(event):
plt.disconnect(plt.connect('motion_notify_event', on_move))
t = np.arange(0.0, 1.0, 0.01)
s = [1.2 for i in t]
s1 = [3 for i in t]
fig, ax = plt.subplots()
ax.plot(t, s)
ax.plot(t,s1)
plt.connect('button_press_event', on_press)
plt.connect('button_release_event',on_button_release)
plt.show()
2. 多条图线的修改
待补充!有好的建议可以和俺说,俺一定好好学习,好好请教。