使用matplotlib的手动图例，使用plot和fill_between的混合

Question

I am plotting the results of a model fit to data with matplotlib. I plot my model results with a colored line, and I plot the data using fill_between so that the error band is shown. I want my legend to display a gray error band and a gray solid line , even though the curve and band in the figure are plotted in different colors.

I know how to accomplish an independently defined legend using a combination of ax.add_artist and matplotlib.lines , but I do not know how to add an error band using this trick.

import numpy as np 
from matplotlib.pylab import plt
from matplotlib import lines as mlines

x = np.linspace(0, 1, 100)
y = x + 1
ylow, yhigh = 0.9*np.exp(x), 1.1*np.exp(x)

fig, ax = plt.subplots(1, 1)
curve1 = ax.plot(x, y, color='red')
band1 = ax.fill_between(x, ylow, yhigh, color='blue')

solid_line = mlines.Line2D([], [], ls='-', c='gray', label='gray-colored label for line')
band = mlines.Line2D([], [], ls='-', c='gray', label='(incorrect) gray-colored label for band')
first_legend = plt.legend(handles=[solid_line, band], loc=2)
ax.add_artist(first_legend)

Question: how can I get my legend to show a gray band for band1 and a gray line for curve1 ?

Answer 1

You don't have to plot new lines just for legend. You can (and probably should) use your plot with label. Once you create your legend that way, you can change the color of the legendHandles . I hope this is what you want:

import numpy as np 
from matplotlib.pylab import plt
from matplotlib import lines as mlines

x = np.linspace(0, 1, 100)
y = x + 1
ylow, yhigh = 0.9*np.exp(x), 1.1*np.exp(x)

fig, ax = plt.subplots(1, 1)
curve1 = ax.plot(x, y, color='red', label='gray-colored label for line')
band1 = ax.fill_between(x, ylow, yhigh, color='blue', label='(incorrect) gray-colored label for band')

leg = ax.legend(loc=2)
leg.legendHandles[0].set_color('gray')
leg.legendHandles[1].set_color('gray')
plt.show()