So platzieren Sie einzelne Tags für ein Streudiagramm

214

Ich versuche, ein Streudiagramm in matplotlib zu erstellen, und konnte keine Möglichkeit finden, den Punkten Tags hinzuzufügen. Beispielsweise:

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

Ich möchte, dass die Punkte in "y" Beschriftungen wie "Punkt 1", "Punkt 2" usw. haben. Ich konnte es nicht herausfinden.

python matplotlib J. Velazquez-Muriel
quelle

Antworten:

374

Verwenden Sie möglicherweise 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()

Geben Sie hier die Bildbeschreibung ein

unutbu
quelle
8
Das wollte ich sagen. Link zu Dokumenten: matplotlib.sourceforge.net/api/… Link zur Demo: matplotlib.sourceforge.net/examples/pylab_examples/…
Paul
4
@ubuntu Ist es möglich, Zahlen (oder Beschriftungen) anstelle der Punkte zu verwenden?
Vladtn
@Vladtn: Sie können die Anmerkungen ändern, indem Sie die labelsVariable neu definieren.
Unutbu
16
@Vladtn: Sie können die Kreise entfernen, indem Sie sie weglassen plt.scatter. Mit können Sie beliebigen Text auf dem Bild platzieren plt.annotate(label, xy = (x, y), xytext = (0, 0), textcoords = 'offset points'). Hinweis xytext = (0, 0)bedeutet keinen Versatz und das Weglassen arrowpropsführt plt.annotatedazu, dass kein Pfeil gezeichnet wird.
Unutbu
1
@OParker: Wechseln 'point{0}'.format(i)zu 'point{0}'.format(i+1). Oder Sie können das range: ['point{0}'.format(i) for i in range(N)]in ändern ['point{0}'.format(i) for i in range(1,N+1)].
Unutbu