augmentation de la largeur de raie de la légende des lignes dans matplotlib

Je sais que si je change la largeur de raie d'une ligne, qui est automatiquement mis à jour dans la légende.
Cependant, je voudrais juste changer la légende de la largeur de raie sans affecter l'intrigue.

InformationsquelleAutor Andrea Zonca | 2012-03-14
  1. 59

    Voici un exemple simple de la façon de le faire:

    import numpy as np
import matplotlib.pyplot as plt

# make some data
x = np.linspace(0, 2*np.pi)
y1 = np.sin(x)
y2 = np.cos(x)

# plot sin(x) and cos(x)
p1 = plt.plot(x, y1, 'b-', linewidth=1.0)
p2 = plt.plot(x, y2, 'r-', linewidth=1.0)

# make a legend for both plots
leg = plt.legend([p1, p2], ['sin(x)', 'cos(x)'], loc=1)

# set the linewidth of each legend object
for legobj in leg.legendHandles:
    legobj.set_linewidth(2.0)

plt.show()
    InformationsquelleAutor Brendan Wood
  2. 5

    @Brendan Wood utilisation d'une méthode de l'api fournie par pyplot. Dans matplotlib, le style orienté objet à l'aide d'axes est préférée. Voici comment vous pouvez atteindre cet objectif en utilisant axes méthode.

    import numpy as np
import matplotlib.pyplot as plt

# make some data
x = np.linspace(0, 2*np.pi)
y1 = np.sin(x)
y2 = np.cos(x)

fig, ax = plt.subplots()
ax.plot(x, y1, linewidth=1.0, label='sin(x)')
ax.plot(x, y2, linewidth=1.0, label='cos(x)')
leg = ax.legend()

for line in leg.get_lines():
    line.set_linewidth(4.0)

plt.show()

    Le produit de la parcelle est indiqué ci-dessous,
    augmentation de la largeur de raie de la légende des lignes dans matplotlib

    InformationsquelleAutor jdhao
  3. 3

    Si vous souhaitez modifier toutes les lignes dans une parcelle, il peut être utile de définir votre propre légende gestionnaire:

    import matplotlib.pyplot as plt
from matplotlib import legend_handler
from matplotlib.lines import Line2D
import numpy as np
class MyHandlerLine2D(legend_handler.HandlerLine2D):
def create_artists(self, legend, orig_handle,
xdescent, ydescent, width, height, fontsize,
trans):
xdata, xdata_marker = self.get_xdata(legend, xdescent, ydescent,
width, height, fontsize)
ydata = ((height-ydescent)/2.)*np.ones(xdata.shape, float)
legline = Line2D(xdata, ydata)
self.update_prop(legline, orig_handle, legend)
#legline.update_from(orig_handle)
#legend._set_artist_props(legline) # after update
#legline.set_clip_box(None)
#legline.set_clip_path(None)
legline.set_drawstyle('default')
legline.set_marker("")
legline.set_linewidth(10)
legline_marker = Line2D(xdata_marker, ydata[:len(xdata_marker)])
self.update_prop(legline_marker, orig_handle, legend)
#legline_marker.update_from(orig_handle)
#legend._set_artist_props(legline_marker)
#legline_marker.set_clip_box(None)
#legline_marker.set_clip_path(None)
legline_marker.set_linestyle('None')
if legend.markerscale != 1:
newsz = legline_marker.get_markersize()*legend.markerscale
legline_marker.set_markersize(newsz)
# we don't want to add this to the return list because
# the texts and handles are assumed to be in one-to-one
# correpondence.
legline._legmarker = legline_marker
return [legline, legline_marker]
plt.plot( [0, 1], [0, 1], '-r', lw=1, label='Line' )
plt.legend(handler_map={Line2D:MyHandlerLine2D()})
plt.show()
    InformationsquelleAutor David Zwicker
  4. 3

    Par défaut, la légende contient les lignes elles-mêmes. Par conséquent, la modification de la largeur de raie des lignes dans la toile va également changer le nombre de lignes dans la légende (et vice-versa, car ils sont essentiellement le même objet).

    Une solution possible est d'utiliser une copie de l'artiste de la toile et le changement de la copie de la largeur de raie.

    import numpy as np
import matplotlib.pyplot as plt
import copy
x = np.linspace(0, 2*np.pi)
y1 = np.sin(x)
y2 = np.cos(x)
fig = plt.figure()
ax  = fig.add_subplot(111)
ax.plot(x, y1, c='b', label='y1',linewidth=1.0)
ax.plot(x, y2, c='r', label='y2')
# obtain the handles and labels from the figure
handles, labels = ax.get_legend_handles_labels()
# copy the handles
handles = [copy.copy(ha) for ha in handles ]
# set the linewidths to the copies
[ha.set_linewidth(7) for ha in handles ]
# put the copies into the legend
leg = plt.legend(handles=handles, labels=labels)
plt.savefig('leg_example')
plt.show()

    augmentation de la largeur de raie de la légende des lignes dans matplotlib

    Une autre option serait d'utiliser un handler_map et la mise à jour de la fonction. C'est en quelque sorte automatique, en précisant le gestionnaire de la carte ferait automatiquement une ligne dans la légende de 7 points de large.

    import numpy as np
import matplotlib.pyplot as plt
from matplotlib.legend_handler import HandlerLine2D
x = np.linspace(0, 2*np.pi)
y1 = np.sin(x)
y2 = np.cos(x)
fig = plt.figure()
ax  = fig.add_subplot(111)
ax.plot(x, y1, c='b', label='y1',linewidth=1.0)
ax.plot(x, y2, c='r', label='y2')
linewidth=7
def update(handle, orig):
handle.update_from(orig)
handle.set_linewidth(7)
plt.legend(handler_map={plt.Line2D : HandlerLine2D(update_func=update)})
plt.show()

    Le résultat est le même que ci-dessus.

    • nice, j'étais à la recherche de cette plus tôt. Espérons qu'il obtient accepté ici ;]
    • J'ai mis à jour cette réponse avec une option différente.
    InformationsquelleAutor ImportanceOfBeingErnest