Dodawanie arbitralnej linii do wykresu matplotlib w ipythonie notebooka

Question

Jestem raczej nowy zarówno dla Pythona/Matplotlib, jak i korzystania z niego przez notatnik ipython. Próbuję dodać kilka linii adnotacji do istniejącego wykresu i nie mogę wymyślić, jak renderować linie na wykresie. Tak więc, na przykład, jeśli wykreślić następujące:

import numpy as np 
np.random.seed(5) 
x = arange(1, 101) 
y = 20 + 3 * x + np.random.normal(0, 60, 100) 
p = plot(x, y, "o") 

otrzymuję następujący wykres:

tak jak bym dodać pionową linię z (70,100) do (70250) ? A co z linią ukośną od (70,100) do (90,200)?

Próbowałem już kilku rzeczy z Line2D(), powodując tylko pomyłkę z mojej strony. W R po prostu użyłbym funkcji segments(), która dodawałaby segmenty linii. Czy istnieje odpowiednik w matplotlib?

Answer 1

Można bezpośrednio wykreślić linie chcesz karmiąc polecenia plot z odpowiednimi danymi (granice segmentów) :

plot([x1, x2], [y1, y2], color='k', linestyle='-', linewidth=2)

(oczywiście można wybrać kolor, szerokość linii, styl linii, itd.)

ze swojego przykład:

import numpy as np 
import matplotlib.pyplot as plt 

np.random.seed(5) 
x = np.arange(1, 101) 
y = 20 + 3 * x + np.random.normal(0, 60, 100) 
plt.plot(x, y, "o") 


# draw vertical line from (70,100) to (70, 250) 
plt.plot([70, 70], [100, 250], 'k-', lw=2) 

# draw diagonal line from (70, 90) to (90, 200) 
plt.plot([70, 90], [90, 200], 'k-') 

plt.show() 

Answer 2

Korzystanie vlines:

import numpy as np 
np.random.seed(5) 
x = arange(1, 101) 
y = 20 + 3 * x + np.random.normal(0, 60, 100) 
p = plot(x, y, "o") 
vlines(70,100,250) 

The podstawowych podpisów zaproszeniem:

vlines(x, ymin, ymax) 
hlines(y, xmin, xmax) 

Answer 3

Jeszcze nie jest za późno na przybyszów.

plt.axvline(x, color='r') 

http://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.axvline

zajmuje zakres Y, jak również za pomocą Ymin i ymax.

Answer 4

Matplolib pozwala teraz na "linie adnotacji", których szukał OP. Funkcja annotate() umożliwia kilka postaci ścieżek łączenia, a jednym z nich jest strzałka bez głowy i bez końca, tj. Prosta linia.

ax.annotate("", 
      xy=(0.2, 0.2), xycoords='data', 
      xytext=(0.8, 0.8), textcoords='data', 
      arrowprops=dict(arrowstyle="-", 
         connectionstyle="arc3, rad=0"), 
      ) 

W the documentation mówi można wyciągnąć tylko strzałę z pustym ciągiem jako pierwszy argument.

Od przykład OP:

%matplotlib notebook 
import numpy as np 
import matplotlib.pyplot as plt 

np.random.seed(5) 
x = np.arange(1, 101) 
y = 20 + 3 * x + np.random.normal(0, 60, 100) 
plt.plot(x, y, "o") 


# draw vertical line from (70,100) to (70, 250) 
plt.annotate("", 
       xy=(70, 100), xycoords='data', 
       xytext=(70, 250), textcoords='data', 
       arrowprops=dict(arrowstyle="-", 
           connectionstyle="arc3,rad=0."), 
      ) 

# draw diagonal line from (70, 90) to (90, 200) 
plt.annotate("", 
       xy=(70, 90), xycoords='data', 
       xytext=(90, 200), textcoords='data', 
       arrowprops=dict(arrowstyle="-", 
           connectionstyle="arc3,rad=0."), 
      ) 

plt.show() 

Podobnie jak w podejściu w odpowiedzi gcalmettes, można wybrać kolor, grubość linii, styl linii, itp ..

Oto zmiana części kodu, która spowodowałaby, że jedna z dwóch przykładowych linii jest czerwona, szersza, a nie w 100% nieprzejrzysta.

# draw vertical line from (70,100) to (70, 250) 
plt.annotate("", 
       xy=(70, 100), xycoords='data', 
       xytext=(70, 250), textcoords='data', 
       arrowprops=dict(arrowstyle="-", 
           edgecolor = "red", 
           linewidth=5, 
           alpha=0.65, 
           connectionstyle="arc3,rad=0."), 
      ) 

Można również dodać krzywą linią łączącą regulując connectionstyle.