matplotlib: jak opisać punkt na scatter automatycznie umieszczonej strzałki?

Question

jeśli zrobię wykres punktowy z matplotlib:

plt.scatter(randn(100),randn(100)) 
# set x, y lims 
plt.xlim([...]) 
plt.ylim([...]) 

chciałbym opisywać dany punkt (x, y) ze strzałką skierowaną do niego i etykiety. Wiem, że można to zrobić za pomocą annotate, ale chciałbym, aby strzałka i jej etykieta były umieszczone "optymalnie" w taki sposób, aby jeśli to możliwe (biorąc pod uwagę obecne osie/ograniczenia), strzałka i etykieta nie pokrywają się z innymi punktami. np. jeśli chcesz oznaczyć punkt odstający. czy jest jakiś sposób na zrobienie tego? nie musi to być doskonałe, ale tylko inteligentne umieszczenie strzałki/etykiety, z podaniem jedynie współrzędnych punktu, który ma być oznaczony. dzięki.

Answer 1

Zasadniczo nie, nie ma.

Silniki układu, które obsługują umieszczanie podobnych etykiet na mapie, są zaskakująco skomplikowane i wykraczają poza zakres matplotlib. (Skrzyżowania skrzynek z ograniczeniami w rzeczywistości są raczej kiepskim sposobem decydowania o tym, gdzie umieścić etykiety.) Jaki sens ma napisanie tony kodu dla czegoś, co działa tylko w jednym przypadku na 1000?)

Inne niż z powodu ilość złożonego renderowania tekstu wykonywanego przez matplotlib (np. latex), niemożliwe jest określenie zakresu tekstu bez pełnego renderowania go (co jest raczej powolne).

Jednak w wielu przypadkach okaże się, że użycie przezroczystego pudełka za etykietą umieszczoną z adnotacją jest odpowiednim rozwiązaniem.

E.g.

import numpy as np 
import matplotlib.pyplot as plt 

np.random.seed(1) 
x, y = np.random.random((2,500)) 

fig, ax = plt.subplots() 
ax.plot(x, y, 'bo') 

# The key option here is `bbox`. I'm just going a bit crazy with it. 
ax.annotate('Something', xy=(x[0], y[0]), xytext=(-20,20), 
      textcoords='offset points', ha='center', va='bottom', 
      bbox=dict(boxstyle='round,pad=0.2', fc='yellow', alpha=0.3), 
      arrowprops=dict(arrowstyle='->', connectionstyle='arc3,rad=0.5', 
          color='red')) 

plt.show() 

Answer 2

Zastosowanie adjustText (pełne ujawnienie, napisałem go).

Oznaczmy etykietą pierwsze 10 punktów. Jedynym parametrem, który zmieniłem, było obniżenie siły odpychania z punktów, ponieważ jest ich tak dużo i chcemy, aby algorytm zajmował nieco więcej czasu i dokładniej umieszczał adnotacje.

import numpy as np 
import matplotlib.pyplot as plt 
from adjustText import adjust_text 
np.random.seed(1) 
x, y = np.random.random((2,500)) 

fig, ax = plt.subplots() 
ax.plot(x, y, 'bo') 
ts = [] 
for i in range(10): 
    ts.append(plt.text(x[i], y[i], 'Something'+str(i))) 
adjust_text(ts, x=x, y=y, force_points=0.1, arrowprops=dict(arrowstyle='->', 
color='red')) 
plt.show() 

To nie jest idealne, ale punkty są tu naprawdę gęsta i czasami nie ma sposobu, aby umieścić tekst blisko swojego celu bez nakładania żadnego z nich. Ale to wszystko jest automatyczne i łatwe w użyciu, a także nie pozwala na nakładanie się etykiet.

PS Posługuje się prostymi przecięciami, ale raczej powiem!