首页 > 解决方案 > python matplotlib 自定义图例

问题描述

我正在使用 python 将一些传感器数据从 arduino 流式传输并绘制到我的电脑。我希望图表有一个带有一些值的图例(当然,我也从我的串行连接中流式传输)但在所有示例中,我发现图例的左侧有一个很大的空间。

我怎样才能删除它?(或使用其他方法获得相同的输出)

这是我的代码:

#!/usr/bin/python

import matplotlib.pyplot as plt
import time
import numpy as np
from scipy.interpolate import spline
import matplotlib.patches as mpatches

# Local variables
x = []
y = []

# Open the data file for reading lines
datafile = open('./plotfiles/forceplot.txt', 'r')
sepfile = datafile.read().split('\n')
datafile.close()

# Create a canvas to place the subgraphs
canvas = plt.figure()
rect = canvas.patch
rect.set_facecolor('white')

# Iterate through the lines and parse them
for datapair in sepfile:
    if datapair:
        xypair = datapair.split(',')
        x.append(int(xypair[0]))
        y.append(int(xypair[1]))

x_sm = np.array(x)
y_sm = np.array(y)

x_smooth = np.linspace(x_sm.min(), x_sm.max(), 200)
y_smooth = spline(x, y, x_smooth)

# Define the matrix of 1x1 to place subplots
# Placing the plot1 on 1x1 matrix, at pos 1
sp1 = canvas.add_subplot(1,1,1)
#sp1.plot(x, y, 'red', linewidth=2)
sp1.plot(x_smooth, y_smooth, 'red', linewidth=1)

# Colorcode the tick tabs 
sp1.tick_params(axis='x', colors='red')
sp1.tick_params(axis='y', colors='red')

# Colorcode the spine of the graph
sp1.spines['bottom'].set_color('r')
sp1.spines['top'].set_color('r')
sp1.spines['left'].set_color('r')
sp1.spines['right'].set_color('r')

# Put the title and labels
sp1.set_title('matplotlib example 1', color='red')
sp1.set_xlabel('matplot x label', color='red')
sp1.set_ylabel('matplot y label', color='red')

# Show the plot/image
time = 'time: 00:23'
s = 3
sh = "success hits: "+str(s)
legend_dict = { sh: sh, 'total hits' : '5', time : time}
patchList=[]
for key in legend_dict:
    data_key = mpatches.Patch(color='None', label=key)
    patchList.append(data_key)

plt.legend(handles=patchList)

plt.tight_layout()
plt.grid(alpha=0.8)
plt.savefig("example6.eps")
plt.show()

这是输出:

在此处输入图像描述

标签: pythonpython-3.xmatplotliblegend

解决方案


由于颜色为无,因此显示空白。如果您使用文本:

text(x, y,'time: 00:23\nsuccess hits: 3\ntotal hits: 5', horizo​​ntalalignment='center', verticalalignment='center')


推荐阅读