Matplotlib手绘曲线
代码：
from matplotlib import pyplot as plt
# 支持中文
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
class LineDrawer(object):
def __init__(self, line, axis, figure): # 将line(Line2D(matplotlib.artist.Artist)作为该类的一个属性，该类只做好数据处理与传递工作
self.line = line
self.axis = axis
self.figure = figure
# 用于拖拽坐标(AxesSubplot(SubplotBase, matplotlib.axes._axes.Axes))
self.press = None
self.cur_xlim = None
self.cur_ylim = None
self.x0 = None
self.y0 = None
self.x1 = None
self.y1 = None
self.xpress = None
self.ypress = None
# 用于将坐标点的存储
self.lines_x_list = [] # 各分段的总数据
self.lines_y_list = [] # 各分段的总数据
# 撤回的回收线段
self.undo_lines_x_list = []
self.undo_lines_y_list = []
#self.xs = list(line.get_xdata()) # xs属于自定义变量名
#self.ys = list(line.get_ydata())
self.cid_press = self.line.figure.canvas.mpl_connect('button_press_event', self.onPress) # 按下鼠标三键
self.cid_release = self.line.figure.canvas.mpl_connect('button_release_event', self.onRelease) # 释放鼠标三键
#self.cid_release = self.line.figure.canvas.mpl_connect('scroll_event', self.zoom_factory) # 滚轮
def Get_lines_xy_show_From_lines_xy_list(self):
#print(len(self.lines_x_list), len(self.lines_y_list))
self.lines_x_show = [] # 取出已经画好的n段曲线
self.lines_y_show = [] # 取出已经画好的n段曲线
for l in self.lines_x_list:
for x in l:
self.lines_x_show.append(x)
for l in self.lines_y_list:
for y in l:
self.lines_y_show.append(y)
def onPress(self, event):
if event.button == 3: # 鼠标右键 手绘图
self.cid_motion = self.figure.canvas.mpl_connect("motion_notify_event", self.mouse_move)
self.xdata_this_move = []
self.ydata_this_move = []
self.Get_lines_xy_show_From_lines_xy_list()
elif event.button == 2: # 鼠标中键
if not event.dblclick: #单击鼠标中键 撤回(撤销最后一段)
try:
#print(len(self.lines_x_list), len(self.lines_y_list))
self.undo_lines_x_list.append(self.lines_x_list.pop(-1)) # 回收
self.undo_lines_y_list.append(self.lines_y_list.pop(-1)) # 回收
except IndexError:
return
else:#双击鼠标中键 取消撤回
'''
try:
#print(len(self.lines_x_list), len(self.lines_y_list))
self.lines_x_list.append(self.undo_lines_x_list.pop(-1)) # 取消撤回
self.lines_y_list.append(self.undo_lines_y_list.pop(-1)) # 取消撤回
except IndexError:
return
'''
pass
self.Get_lines_xy_show_From_lines_xy_list()
#print(len(self.lines_x_show), len(self.lines_y_show))
self.line.set_data(self.lines_x_show, self.lines_y_show) # 连接各段，全部画出来
#self.line.figure.canvas.get_renderer(cleared=True)
self.line.figure.canvas.draw_idle()
def mouse_move(self, event):
self.xdata_this_move.append(event.xdata)
self.ydata_this_move.append(event.ydata)
self.lines_x_show.append(event.xdata)
self.lines_y_show.append(event.ydata)
self.line.set_data(self.lines_x_show, self.lines_y_show) # 连接各段，全部画出来
self.line.figure.canvas.draw_idle() # 数据重画
def onRelease(self, event):
if event.button == 3: # 鼠标右键 手绘图
self.cid_release = self.figure.canvas.mpl_disconnect(self.cid_motion)
self.lines_x_list.append(self.xdata_this_move) # 释放后添加本次运动的x点
self.lines_y_list.append(self.ydata_this_move) # 释放后添加本次运动的y点
#print(self.xdata_this_move, self.ydata_this_move, '\n==================\n\n')
# 缩放
def zoom_factory(self, base_scale=1.1):
# ax (AxesSubplot(SubplotBase, matplotlib.axes._axes.Axes))
def zoom(event):
cur_xlim = self.axis.get_xlim()
cur_ylim = self.axis.get_ylim()
xdata = event.xdata # get event x location
ydata = event.ydata # get event y location
if event.button == 'down':
# deal with zoom in
scale_factor = 1 / base_scale
elif event.button == 'up':
# deal with zoom out
scale_factor = base_scale
else:
# deal with something that should never happen
scale_factor = 1
#print(event.button)
new_width = (cur_xlim[1] - cur_xlim[0]) * scale_factor
new_height = (cur_ylim[1] - cur_ylim[0]) * scale_factor
relx = (cur_xlim[1] - xdata)/(cur_xlim[1] - cur_xlim[0])
rely = (cur_ylim[1] - ydata)/(cur_ylim[1] - cur_ylim[0])
self.axis.set_xlim([xdata - new_width * (1-relx), xdata + new_width * (relx)])
self.axis.set_ylim([ydata - new_height * (1-rely), ydata + new_height * (rely)])
self.axis.figure.canvas.draw() # 数据重画
#fig = self.axis.get_figure() # get the figure of interest
self.figure.canvas.mpl_connect('scroll_event', zoom)
return zoom
# 左键拖动
def pan_factory(self):
def onPress(event):
if event.button == 1:
if event.inaxes != self.axis: return
self.cur_xlim = self.axis.get_xlim()
self.cur_ylim = self.axis.get_ylim()
self.press = self.x0, self.y0, event.xdata, event.ydata
self.x0, self.y0, self.xpress, self.ypress = self.press
def onRelease(event):
self.press = None
self.axis.figure.canvas.draw() # 数据重画
def onMotion(event):
if self.press is None: return
if event.inaxes != self.axis: return
dx = event.xdata - self.xpress
dy = event.ydata - self.ypress
self.cur_xlim -= dx
self.cur_ylim -= dy
self.axis.set_xlim(self.cur_xlim)
self.axis.set_ylim(self.cur_ylim)
self.axis.figure.canvas.draw() # 数据重画
#fig = self.axis.get_figure() # get the figure of interest
# attach the call back
self.figure.canvas.mpl_connect('button_press_event',onPress)
self.figure.canvas.mpl_connect('button_release_event',onRelease)
self.figure.canvas.mpl_connect('motion_notify_event',onMotion)
#return the function
return onMotion
def main():
l = [500, 657, 814, 971, 1128, 1285, 1442, 1599, 1756, 1913, 2070, 1913, 1756, 1599, 1442, 1285, 1128, 971, 814, 657, 500, 1518, 575, 1725, 500]
X = []
Y = []
n = 0
for i in l:
for j in range(120):
n += 1
X.append(n)
Y.append(i)
fig = plt.figure()
ax = fig.add_subplot(111) # class AxesSubplot(SubplotBase, matplotlib.axes._axes.Axes)
ax.set_title('Matplotlib手绘曲线')
line_fix, = ax.plot(X, Y, lw='0.5')
line_draw_by_hand, = ax.plot([], [], lw=1)
linebuilder = LineDrawer(line=line_draw_by_hand, axis=ax, figure=fig)
figZoom = linebuilder.zoom_factory(base_scale = 1.1)
figPan = linebuilder.pan_factory()
plt.show()
# 关闭图像后, 再执行以下代码
X_line_draw = []
Y_line_draw = []
for l in linebuilder.lines_x_list:
for i in l:
X_line_draw.append(i)
for l in linebuilder.lines_y_list:
for i in l:
Y_line_draw.append(i)
out = open('手绘数据.txt', 'w', encoding='utf8')
for i in range(len(X_line_draw)):
x = X_line_draw[i]
y = Y_line_draw[i]
out.write(f'{x}\t{y}\n')
out.close()
if __name__ == '__main__':
main()
image-20210325221531635.png
手绘图数据
25.638961693548367 546.1584093872231
36.72397513440853 557.8551871856957
50.580241935483855 563.7035760849321
50.580241935483855 560.7793816353139
50.580241935483855 554.9309927360775
64.43650873655906 563.7035760849321
147.57410954301065 657.2777984727138
216.85544354838703 762.5487986589683
214.084190188172 762.5487986589683
208.5416834677419 756.7004097597319
214.084190188172 791.79074315515
244.56797715053756 891.2133544421681
261.1954973118279 917.5311044887317
261.1954973118279 891.2133544421681
297.2217909946236 867.8197988452226
336.01933803763427 929.2278822872045
358.1893649193547 1025.7262991246043
363.7318716397848 1034.498882473459
380.35939180107505 1008.1811324268953
408.0719254032257 1014.0295213261317
424.69944556451594 1028.6504935742225
546.6345934139782 1163.1634382566588
627.0009408602149 1300.6005773887132
657.4847278225806 1326.9183274352768
660.2559811827956 1312.297355187186
665.7984879032257 1312.297355187186
693.5110215053761 1411.719966474204
723.9948084677418 1502.3699944123675
746.1648353494621 1458.5070776680948
779.4198756720428 1402.9473831253495
796.0473958333331 1423.4167442726766
832.0736895161289 1522.8393555596947
895.812516801075 1654.428105792513
912.4400369623653 1669.0490780406037
915.2112903225803 1651.5039113428948
948.4663306451611 1677.8216613894583
1017.7476646505373 1753.850717079531
1120.2840389784942 1873.7426895138763
1167.3953461021501 1914.6814118085308
1178.4803595430103 1879.5910784131127
1192.3366263440857 1870.818495064258
1220.0491599462362 1917.605606258149
1231.1341733870963 1929.3023840566216
1239.4479334677417 1926.3781896070034
1727.1885248655913 1505.2941888619857
1724.417271505376 1484.8248277146586
1724.417271505376 1478.9764388154222
1735.5022849462362 1473.1280499161858
1763.2148185483866 1493.597411063513
1782.613592069892 1443.886105420004
1824.182392473118 1297.676382939095
1838.0386592741934 1318.1457440864224
1879.6074596774188 1373.7054386291677
1907.3199932795696 1332.7667163345131
1943.3462869623654 1291.8279940398586
1932.2612735215048 1280.131216241386
2084.6802083333328 970.1666045818589
2106.8502352150535 920.4552989383499
2112.3927419354836 917.5311044887317
2156.7327956989243 926.3036878375863
2167.8178091397845 926.3036878375863
2228.785383064516 911.6827155894954
2298.066717069892 891.2133544421681
2300.837970430107 894.1375488917863
主页：https://www.zhihu.com/people/caviar126