Menambahkan baris sembarang ke plot matplotlib di notebook ipython

Saya agak baru mengenal python / matplotlib dan menggunakannya melalui notebook ipython. Saya mencoba menambahkan beberapa baris anotasi ke grafik yang ada dan saya tidak tahu cara merender garis pada grafik. Jadi, misalnya, jika saya memplot yang berikut ini:

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")

Saya mendapatkan grafik berikut:

Jadi bagaimana cara menambahkan garis vertikal dari (70.100) hingga (70.250)? Bagaimana dengan garis diagonal dari (70.100) sampai (90.200)?

Saya telah mencoba beberapa hal yang tidak Line2D()menghasilkan apa-apa selain kebingungan di pihak saya. Di Rsaya hanya akan menggunakan fungsi segment () yang akan menambahkan segmen garis. Apakah ada padanannya matplotlib?

Anda dapat langsung memplot garis yang Anda inginkan dengan memberi makan plotperintah dengan data yang sesuai (batas segmen):

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

(tentu saja Anda dapat memilih warna, lebar garis, gaya garis, dll.)

Dari contoh Anda:

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()

Belum terlambat untuk para pendatang baru .

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

Ini mengambil kisaran y juga, menggunakan ymin dan ymax.

Menggunakan 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)

Tanda tangan panggilan dasar adalah:

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

Matplolib sekarang mengizinkan 'baris anotasi' seperti yang dicari OP. The annotate()Fungsi memungkinkan beberapa bentuk menghubungkan jalur dan tanpa kepala dan tailess panah, yaitu, garis sederhana, adalah salah satunya.

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

Dalam dokumentasi dikatakan bahwa Anda hanya dapat menggambar panah dengan string kosong sebagai argumen pertama.

Dari contoh 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()

Seperti pada pendekatan dalam jawaban gcalmettes, Anda dapat memilih warna, lebar garis, gaya garis, dll.

Berikut adalah perubahan pada bagian kode yang akan membuat salah satu dari dua baris contoh menjadi merah, lebih lebar, dan tidak 100% buram.

# 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."), 
              )

Anda juga dapat menambahkan kurva ke garis penghubung dengan menyesuaikan connectionstyle.

Daripada menyalahgunakan plotatau annotate, yang tidak efisien untuk banyak baris, Anda dapat menggunakan matplotlib.collections.LineCollection:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection

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

# Takes list of lines, where each line is a sequence of coordinates
l1 = [(70, 100), (70, 250)]
l2 = [(70, 90), (90, 200)]
lc = LineCollection([l1, l2], color=["k","blue"], lw=2)

plt.gca().add_collection(lc)

plt.show()

Dibutuhkan daftar baris [l1, l2, ...], dimana setiap baris adalah urutan N koordinat ( N bisa lebih dari dua).

Kata kunci pemformatan standar tersedia, menerima nilai tunggal, dalam hal ini nilai berlaku untuk setiap baris, atau urutan M values , dalam hal ini nilai untuk baris ke- i adalah values[i % M].