Come inserire singoli tag per un diagramma a dispersione

214

Sto provando a fare un diagramma a dispersione in matplotlib e non sono riuscito a trovare un modo per aggiungere tag ai punti. Per esempio:

scatter1=plt.scatter(data1["x"], data1["y"], marker="o",
                     c="blue",
                     facecolors="white",
                     edgecolors="blue")

Voglio che i punti in "y" abbiano etichette come "punto 1", "punto 2", ecc. Non sono riuscito a capirlo.

python matplotlib

— J. Velazquez-Muriel
fonte

374

Forse usa plt.annotate :

import numpy as np
import matplotlib.pyplot as plt

N = 10
data = np.random.random((N, 4))
labels = ['point{0}'.format(i) for i in range(N)]

plt.subplots_adjust(bottom = 0.1)
plt.scatter(
    data[:, 0], data[:, 1], marker='o', c=data[:, 2], s=data[:, 3] * 1500,
    cmap=plt.get_cmap('Spectral'))

for label, x, y in zip(labels, data[:, 0], data[:, 1]):
    plt.annotate(
        label,
        xy=(x, y), xytext=(-20, 20),
        textcoords='offset points', ha='right', va='bottom',
        bbox=dict(boxstyle='round,pad=0.5', fc='yellow', alpha=0.5),
        arrowprops=dict(arrowstyle = '->', connectionstyle='arc3,rad=0'))

plt.show()

inserisci qui la descrizione dell'immagine

— unutbu
fonte

Questo è quello che stavo per dire. Link alla documentazione: matplotlib.sourceforge.net/api/… Link alla demo: matplotlib.sourceforge.net/examples/pylab_examples/…

— Paul,

@ubuntu È possibile utilizzare numeri (o etichette) anziché i punti?

— Vladtn,

@Vladtn: è possibile modificare le annotazioni ridefinendo la labelsvariabile.

— unutbu,

@Vladtn: puoi rimuovere i cerchi omettendo plt.scatter. È possibile posizionare testo arbitrario sull'immagine utilizzando plt.annotate(label, xy = (x, y), xytext = (0, 0), textcoords = 'offset points'). Avviso xytext = (0, 0)significa nessun offset e l'omissione arrowpropsprovoca il plt.annotatenon disegnare una freccia.

— unutbu,

@OParker: passare 'point{0}'.format(i)a 'point{0}'.format(i+1). Oppure, è possibile modificare il range: ['point{0}'.format(i) for i in range(N)]in ['point{0}'.format(i) for i in range(1,N+1)].

— unutbu,