Définition d'une taille fixe pour les points dans la légende

J'ai eu un coup d'oeil dans le code source de matplotlib. Mauvaise nouvelle, c'est qu'il ne semble pas y avoir une façon simple de régler l'égalité des tailles de points dans la légende. Il est particulièrement difficile avec des nuages de points (mal: voir la mise à jour ci-dessous). Il y a essentiellement deux alternatives:

1. Changer le maplotlib code
2. Ajouter une transformation dans la PathCollection des objets représentant les points dans l'image. La transformation (mise à l'échelle) a pour prendre la taille en considération.

N'est pas très amusant, bien que #1 semble être plus facile. Le scatter parcelles sont particulièrement difficiles à cet égard.

Cependant, j'ai un hack qui est probablement ce que vous voulez:

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()

Cela donne:

Qui semble être ce que tu voulais.

Les modifications:

• scatter changé en un plotce qui change le marqueur de mise à l'échelle (d'où le sqrt) et rend impossible l'utilisation du changement de marqueur de taille (si c'était prévu)
• le marqueur de changement de la taille manuelle à 6 points pour les deux marqueurs dans la légende

Comme vous pouvez le voir, il utilise caché soulignent les propriétés (_legmarker) et le bug est moche. Il peut se décomposer à une mise à jour de matplotlib.

Mise à jour

Haa, je l'ai trouvé. Un meilleur 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()

Maintenant la _sizes (un autre trait de soulignement de la propriété) fait le tour. Pas besoin de toucher à la source, même si c'est un hack. Mais maintenant, vous pouvez utiliser tout scatter offre.