Einstellung eine Feste Größe für die Punkte in der Legende

Hatte ich einen Blick in den source-code von matplotlib. Schlechte Nachricht ist, dass es scheint nicht zu sein jede einfache Art der Einstellung gleich die Größen der Punkte in der Legende. Besonders schwierig ist es, mit scatter-plots (falsch: siehe update unten). Es gibt im wesentlichen zwei alternativen:

1. Ändern Sie die maplotlib code
2. Fügen Sie eine Transformation in den PathCollection - Objekte, die die Punkte in das Bild. Die Transformation (Skalierung), um die Originale Größe zu berücksichtigen.

Weder von diesen sehr viel Spaß, obwohl #1 scheint einfacher zu sein. Die scatter Grundstücke sind besonders anspruchsvoll in dieser Hinsicht.

Allerdings habe ich das hack, das tut wahrscheinlich das, was Sie wollen:

import matplotlib.pyplot as plt
import numpy as np

def rand_data():
    return np.random.uniform(low=0., high=1., size=(100,))

# Generate data.
x1, y1 = [rand_data() for i in range(2)]
x2, y2 = [rand_data() for i in range(2)]

plt.figure()
plt.plot(x1, y1, 'o', label='first', markersize=np.sqrt(20.), c='b')
plt.plot(x2, y2, 'o', label='second', markersize=np.sqrt(35.), c='r')
# Plot legend.
lgnd = plt.legend(loc="lower left", numpoints=1, fontsize=10)

#change the marker size manually for both lines
lgnd.legendHandles[0]._legmarker.set_markersize(6)
lgnd.legendHandles[1]._legmarker.set_markersize(6)
plt.show()

Gibt es:

Scheint das zu sein, was Sie wollte.

Änderungen:

• scatter geändert in eine plot werden, ändert sich die Markierung Skalierung (daher der sqrt) und macht es unmöglich, zu verwenden ändern der marker-Größe (wenn das beabsichtigt war)
• die marker-Größe manuell geändert werden 6 Punkte für beide Marker in der Legende

Wie Sie sehen können, dieser nutzt versteckte Unterstrich Eigenschaften (_legmarker) und ist bug-hässlich. Es kann brechen und sich bei jedem update in matplotlib.

Update

Haa, ich habe es gefunden. Eine bessere hack:

import matplotlib.pyplot as plt
import numpy as np

def rand_data():
    return np.random.uniform(low=0., high=1., size=(100,))

# Generate data.
x1, y1 = [rand_data() for i in range(2)]
x2, y2 = [rand_data() for i in range(2)]

plt.figure()
plt.scatter(x1, y1, marker='o', label='first', s=20., c='b')
plt.scatter(x2, y2, marker='o', label='second', s=35., c='r')
# Plot legend.
lgnd = plt.legend(loc="lower left", scatterpoints=1, fontsize=10)
lgnd.legendHandles[0]._sizes = [30]
lgnd.legendHandles[1]._sizes = [30]
plt.show()

Nun die _sizes (ein weiterer Unterstrich Eigenschaft) funktioniert der trick. Keine Notwendigkeit, berühren Sie die Quelle, obwohl dies ist ein ziemlich hack. Aber jetzt können Sie alles scatter bietet.

InformationsquelleAutor der Antwort DrV

Hatte ich nicht viel Erfolg mit @DrV - Lösung, aber vielleicht ist mein use-case ist einzigartig. Aufgrund der Dichte der Punkte, bin ich mit der kleinsten Markierung Größe, d.h. plt.plot(x, y, '.', ms=1, ...)und möchten, dass die Legende die Symbole größer.

Folgte ich der Empfehlung, die ich gefunden auf der matplotlib Foren:

1. plot der Daten (keine Etiketten)
2. record-Achsen-limit (xlimits = plt.xlim())
3. Handlung fake-Daten weit Weg von den realen Daten mit Legende-Symbole, Farben und Größen
4. wiederherstellen Achsen-limits (plt.xlim(xlimits))
5. Legende erzeugen

Hier ist, wie es stellte sich heraus (für das die dots sind eigentlich weniger wichtig, dass die Linien):

Hoffe, dies hilft jemand anderes.

InformationsquelleAutor der Antwort Steven C. Howell

Können Sie ein Line2D-Objekt, das ähnelt Ihrem gewählten Marker, außer mit einem anderen marker der Größe Ihrer Wahl, und verwenden Sie, dass die zur Konstruktion der Legende. Das ist nett, weil es nicht erforderlich ist, platzieren Sie ein Objekt in Ihre Achsen (potenziell Triggerung ein resize-Ereignis), und es erfordert nicht die Verwendung von versteckten Parametern. Der einzige wirkliche Nachteil ist, dass Sie die zur Konstruktion der Legende ausdrücklich von Listen der Objekte und Beschriftungen, aber diese ist sehr gut dokumentiert matplotlib-Funktion, so dass es fühlt sich ziemlich sicher zu verwenden.

from matplotlib.lines import Line2D
import matplotlib.pyplot as plt
import numpy as np

def rand_data():
    return np.random.uniform(low=0., high=1., size=(100,))

# Generate data.
x1, y1 = [rand_data() for i in range(2)]
x2, y2 = [rand_data() for i in range(2)]

plt.figure()
plt.scatter(x1, y1, marker='o', label='first', s=20., c='b')
plt.scatter(x2, y2, marker='o', label='second', s=35., c='r')

# Create dummy Line2D objects for legend
h1 = Line2D([0], [0], marker='o', markersize=np.sqrt(20), color='b', linestyle='None')
h2 = Line2D([0], [0], marker='o', markersize=np.sqrt(20), color='r', linestyle='None')

# Set axes limits
plt.gca().set_xlim(-0.2, 1.2)
plt.gca().set_ylim(-0.2, 1.2)

# Plot legend.
plt.legend([h1, h2], ['first', 'second'], loc="lower left", markerscale=2,
           scatterpoints=1, fontsize=10)
plt.show()

Link zum resultierenden Bild

InformationsquelleAutor der Antwort Jeremy Bancroft Brown