Nuage de points de Matplotlib avec légende

Je veux créer un Matplotlib un nuage de points, avec une légende indiquant la couleur de chaque classe. Par exemple, j'ai une liste de x et y valeurs, et une liste de classes valeurs. Chaque élément de la xy et classes listes correspond à un point dans l'intrigue. Je veux que chaque classe d'avoir sa propre couleur, j'en ai déjà codé, mais je veux les classes à être affiché dans une légende. Quels paramètres dois-je passer à la legend() fonction pour y parvenir?

Voici mon code pour l'instant:

x = [1, 3, 4, 6, 7, 9]
y = [0, 0, 5, 8, 8, 8]
classes = ['A', 'A', 'B', 'C', 'C', 'C']
colours = ['r', 'r', 'b', 'g', 'g', 'g']
plt.scatter(x, y, c=colours)

source d'informationauteur Karnivaurus

  1. 12

    D'abord, j'ai le sentiment que vous vouliez utiliser des apostrophes, pas backticks lors de la déclaration de couleurs.

    Pour une légende vous avez besoin de certaines formes comme les classes. Par exemple, le code suivant crée une liste de rectangles appelés recs pour chaque couleur dans class_colours.

    import matplotlib.patches as mpatches

classes = ['A','B','C']
class_colours = ['r','b','g']
recs = []
for i in range(0,len(class_colours)):
    recs.append(mpatches.Rectangle((0,0),1,1,fc=class_colours[i]))
plt.legend(recs,classes,loc=4)

    Nuage de points de Matplotlib avec légende

    Vous pouvez utiliser les cercles aussi, si vous le souhaitez, il suffit de cocher la matplotlib.patches de la documentation. Il y a une deuxième façon de créer une légende, dans laquelle vous indiquez le "Label" pour un ensemble de points en utilisant un éparpillement de commande pour chaque jeu. Un exemple est donné ci-dessous.

    classes = ['A','A','B','C','C','C']
colours = ['r','r','b','g','g','g']
for (i,cla) in enumerate(set(classes)):
    xc = [p for (j,p) in enumerate(x) if classes[j]==cla]
    yc = [p for (j,p) in enumerate(y) if classes[j]==cla]
    cols = [c for (j,c) in enumerate(colours) if classes[j]==cla]
    plt.scatter(xc,yc,c=cols,label=cla)
plt.legend(loc=4)

    Nuage de points de Matplotlib avec légende

    La première méthode est la méthode que j'ai personnellement utilisé, la seconde je l'ai trouvé en regardant les matplotlib de la documentation. Depuis les légendes ont été couvrant les points de données, j'ai déménagé, et les emplacements pour les légendes peuvent être trouvés ici. Si il y a une autre façon de faire une légende, je n'étais pas capable de le trouver après quelques recherches rapides dans les docs.

  2. 5

    Il y a deux façons de le faire. L'un d'eux vous donne les entrées de légende pour chaque chose vous intrigue, et l'autre permet de mettre ce que vous voulez dans la légende, voler fortement de cette réponse.

    Voici le premier moyen:

    import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-1,1,100)

fig = plt.figure()
ax = fig.add_subplot(1,1,1)

#Plot something
ax.plot(x,x, color='red', ls="-", label="$P_1(x)$")
ax.plot(x,0.5 * (3*x**2-1), color='green', ls="--", label="$P_2(x)$")
ax.plot(x,0.5 * (5*x**3-3*x), color='blue', ls=":", label="$P_3(x)$")

ax.legend()
plt.show()

    Nuage de points de Matplotlib avec légende

    La ax.legend() fonction a plus d'un usage, le premier crée la légende sur la base des lignes dans axes objet, le second allwos vous pour contrôler les entrées manuellement, et est décrit ici.

    En gros, vous avez besoin pour donner à la légende de la ligne de poignées, et les labels associés.

    L'autre vous permet de mettre ce que vous voulez dans la légende, par la création de la Artist les objets et les étiquettes, et de les transmettre à la ax.legend() fonction. Vous pouvez soit l'utiliser pour seulement certains de vos lignes dans la légende, ou vous pouvez l'utiliser pour mettre ce que vous voulez dans la légende.

    import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-1,1,100)

fig = plt.figure()
ax = fig.add_subplot(1,1,1)

#Plot something
p1, = ax.plot(x,x, color='red', ls="-", label="$P_1(x)$")
p2, = ax.plot(x,0.5 * (3*x**2-1), color='green', ls="--", label="$P_2(x)$")
p3, = ax.plot(x,0.5 * (5*x**3-3*x), color='blue', ls=":", label="$P_3(x)$")

#Create legend from custom artist/label lists
ax.legend([p1,p2], ["$P_1(x)$", "$P_2(x)$"])

plt.show()

    Nuage de points de Matplotlib avec légende

    Ou ici, nous créons de nouvelles Line2D objets, et de leur donner la légende.

    import matplotlib.pyplot as pltit|delete|flag
import numpy as np
import matplotlib.patches as mpatches

x = np.linspace(-1,1,100)

fig = plt.figure()
ax = fig.add_subplot(1,1,1)

#Plot something
p1, = ax.plot(x,x, color='red', ls="-", label="$P_1(x)$")
p2, = ax.plot(x,0.5 * (3*x**2-1), color='green', ls="--", label="$P_2(x)$")
p3, = ax.plot(x,0.5 * (5*x**3-3*x), color='blue', ls=":", label="$P_3(x)$")

fakeLine1 = plt.Line2D([0,0],[0,1], color='Orange', marker='o', linestyle='-')
fakeLine2 = plt.Line2D([0,0],[0,1], color='Purple', marker='^', linestyle='')
fakeLine3 = plt.Line2D([0,0],[0,1], color='LightBlue', marker='*', linestyle=':')

#Create legend from custom artist/label lists
ax.legend([fakeLine1,fakeLine2,fakeLine3], ["label 1", "label 2", "label 3"])

plt.show()

    Nuage de points de Matplotlib avec légende

    J'ai aussi essayé d'obtenir de la méthode à l'aide de patches de travailler, sur le matplotlib légende page du guide, mais il n'a pas l'air de fonctionner donc j'ai renoncé.

  3. 1

    Dans mon projet,j'ai aussi envie de créer un vide d'éparpillement de la légende.Voici ma solution:

    from mpl_toolkits.basemap import Basemap
#use the scatter function from matplotlib.basemap
#you can use pyplot or other else.
select = plt.scatter([], [],s=200,marker='o',linewidths='3',edgecolor='#0000ff',facecolors='none',label=u'监测站点') 
plt.legend(handles=[select],scatterpoints=1)

    Prendre soin de "label","scatterpoints"ci-dessus.