산점도를 위해 개별 태그를 넣는 방법


214

matplotlib에서 산점도를 만들려고하는데 포인트에 태그를 추가하는 방법을 찾을 수 없습니다. 예를 들면 다음과 같습니다.

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

"y"의 포인트에 "point 1", "point 2"등의 레이블을 지정하고 싶습니다. 알아낼 수 없었습니다.

답변:


374

아마도 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()

여기에 이미지 설명을 입력하십시오


8
그것이 제가 말하려고하는 것입니다. 문서 링크 : matplotlib.sourceforge.net/api/… 데모 링크 : matplotlib.sourceforge.net/examples/pylab_examples/…
Paul

4
@ubuntu 포인트 대신 숫자 (또는 레이블)를 사용할 수 있습니까?
Vladtn

@Vladtn : labels변수 를 재정 의하여 주석을 변경할 수 있습니다 .
unutbu

16
@Vladtn :을 생략하여 서클을 삭제할 수 있습니다 plt.scatter. 를 사용하여 이미지에 임의의 텍스트를 배치 할 수 있습니다 plt.annotate(label, xy = (x, y), xytext = (0, 0), textcoords = 'offset points'). 통지 xytext = (0, 0)는 오프셋이 없음을 의미하며 생략 arrowprops하면 plt.annotate화살표가 그려지지 않습니다.
unutbu

1
@OParker : 변경 'point{0}'.format(i)'point{0}'.format(i+1). 또는 range: ['point{0}'.format(i) for i in range(N)]를로 변경할 수 있습니다 ['point{0}'.format(i) for i in range(1,N+1)].
unutbu
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.