Vẽ một đường ngang bằng matplotlib

Tôi đã sử dụng phép nội suy spline để làm mịn chuỗi thời gian và cũng muốn thêm một đường ngang vào cốt truyện. Nhưng dường như có một vấn đề nằm ngoài tầm kiểm soát của tôi. Bất kỳ trợ giúp sẽ thực sự hữu ích. Đây là những gì tôi có:

annual = np.arange(1,21,1)
l = np.array(value_list) # a list with 20 values
spl = UnivariateSpline(annual,l)
xs = np.linspace(1,21,200)
plt.plot(xs,spl(xs),'b')

plt.plot([0,len(xs)],[40,40],'r--',lw=2)
pylab.ylim([0,200])
plt.show()

vấn đề dường như là với việc tôi sử dụng [0,len(xs)]cho âm mưu đường ngang.

Bạn là chính xác, tôi nghĩ rằng [0,len(xs)]đang ném bạn đi. Bạn sẽ muốn sử dụng lại biến trục x gốc xsvà vẽ đồ thị với một mảng numpy khác có cùng độ dài có biến của bạn trong đó.

annual = np.arange(1,21,1)
l = np.array(value_list) # a list with 20 values
spl = UnivariateSpline(annual,l)
xs = np.linspace(1,21,200)
plt.plot(xs,spl(xs),'b')

#####horizontal line
horiz_line_data = np.array([40 for i in xrange(len(xs))])
plt.plot(xs, horiz_line_data, 'r--') 
###########plt.plot([0,len(xs)],[40,40],'r--',lw=2)
pylab.ylim([0,200])
plt.show()

Hy vọng rằng sẽ khắc phục vấn đề!

Bạn đang tìm kiếm axhline(một đường trục ngang). Ví dụ: phần sau đây sẽ cung cấp cho bạn một đường ngang tại y = 0.5:

import matplotlib.pyplot as plt
plt.axhline(y=0.5, color='r', linestyle='-')
plt.show()

Nếu bạn muốn vẽ một đường ngang trong các trục, bạn cũng có thể thử ax.hlines()phương pháp. Bạn cần phải xác định yvị trí và xminvà xmaxtrong các dữ liệu phối hợp (ví dụ, số liệu thực tế của bạn dao động trong trục x). Một đoạn mã mẫu là:

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(1, 21, 200)
y = np.exp(-x)

fig, ax = plt.subplots()
ax.plot(x, y)
ax.hlines(y=0.2, xmin=4, xmax=20, linewidth=2, color='r')

plt.show()

Đoạn mã trên sẽ vẽ một đường ngang trong các trục tại y=0.2. Đường ngang bắt đầu tại x=4và kết thúc tại x=20. Hình ảnh được tạo là:

Sử dụng matplotlib.pyplot.hlines:

• y có thể được thông qua dưới dạng một vị trí: y=40
• y có thể được thông qua dưới dạng nhiều địa điểm: y=[39, 40, 41]
• Nếu bạn đang vẽ một hình với một cái gì đó như thế fig, ax = plt.subplots(), thì thay thế plt.hlineshoặc plt.axhlinebằng ax.hlineshoặc ax.axhline, tương ứng.
• matplotlib.pyplot.axhlinechỉ có thể vẽ một vị trí duy nhất (ví dụ y=40)

import numpy as np
import matplotlib.pyplot as plt

xs = np.linspace(1, 21, 200)
plt.hlines(y=40, xmin=0, xmax=len(xs), colors='r', linestyles='--', lw=2)
plt.show()

Ngoài các câu trả lời upvoted nhất ở đây, người ta cũng có thể chuỗi axhlinesau khi gọi plotvề một pandas's DataFrame.

import pandas as pd

(pd.DataFrame([1, 2, 3])
   .plot(kind='bar', color='orange')
   .axhline(y=1.5));

Một cách hay và dễ dàng cho những người luôn quên lệnh axhlinelà như sau

plt.plot(x, [y]*len(x))

Trong trường hợp của bạn xs = xvà y = 40. Nếu len (x) lớn, thì điều này trở nên không hiệu quả và bạn thực sự nên sử dụng axhline.

Bạn có thể sử dụng plt.gridđể vẽ một đường ngang.

import numpy as np
from matplotlib import pyplot as plt
from scipy.interpolate import UnivariateSpline
from matplotlib.ticker import LinearLocator

# your data here
annual = np.arange(1,21,1)
l = np.random.random(20)
spl = UnivariateSpline(annual,l)
xs = np.linspace(1,21,200)

# plot your data
plt.plot(xs,spl(xs),'b')

# horizental line?
ax = plt.axes()
# three ticks:
ax.yaxis.set_major_locator(LinearLocator(3))
# plot grids only on y axis on major locations
plt.grid(True, which='major', axis='y')

# show
plt.show()