Использование прокси-художника внутри легенды, matplotlib, Python

Question

Вот фрагмент моего кода:

fig2 = plt.figure(figsize=(8,6))
ax1 = fig2.add_subplot(111)
ax1.scatter((logngal),(logm200),c='r',label='$0.0<z<1.0$')
ax1.plot((logngal),(curve_y_1),'y',linewidth=2,label='$slope=%s \pm %s$'%(slope1,slope1_err))


ax1.fill_between(x_pred, lower, upper, color='#888888', alpha=0.5)
p1 = mpatches.Rectangle((0, 0), 1, 1, fc="#888888",alpha=0.5)
ax1.legend([p1],['$1\sigma\/confidence\/limts$'])

fig2.show()

Когда я выполняю вышеизложенное, я вижу только $1\sigma\/confidence\/limts$упоминается в легенде.

В то время как вы можете видеть, что я также называю label='$0.0<z<1.0$' а также label='$slope=%s \pm %s$'%(slope1,slope1_err) в ax1.scatter а также ax1.plot соответственно.

Это не отражено в легенде.

Как добавить все три надписи внутри легенды?

Answer 1

Вам нужно схватить scatter а также plot художников, как вы их сюжет, а затем накормить ручки и метки от них к legend, Например, вот ваш код изменен (с некоторыми примерами данных в начале, чтобы запустить его):

plt.plot возвращает список Line2D объекты, так что если вы читаете это как pplot, = plt.plot(...)Вы распаковываете этот список из одного предмета.

Вы можете использовать .get_label() на pplot а также pscat дать ярлыки легенде.

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as mpatches

# Some things to make your script run when I don't have your data
slope1,slope1_err='this','that'
logngal   = np.linspace(0,1,20)
logm200   = np.random.rand(20)
x_pred    = np.linspace(0,1,20)
curve_y_1 = 0.5*(np.sin(logngal)/2.+np.cos(logngal))
upper     = np.sin(x_pred)/2.
lower     = np.cos(x_pred)
# end of sample data

fig2 = plt.figure(figsize=(8,6))
ax1 = fig2.add_subplot(111)

pscat  = ax1.scatter((logngal),(logm200), c='r',label='$0.0<z<1.0$')
pplot, = ax1.plot((logngal),(curve_y_1),'y',linewidth=2,label='$slope=%s \pm %s$'%(slope1,slope1_err))

ax1.fill_between(x_pred, lower, upper,color='#888888', alpha=0.5)

p1 = mpatches.Rectangle((0, 0), 1, 1, fc="#888888",alpha=0.5)

handles = [p1,pplot,pscat]
labels  = ['$1\sigma\/confidence\/limts$',pplot.get_label(),pscat.get_label()]

ax1.legend(handles,labels)

fig2.show()