Tracez une ligne horizontale en utilisant matplotlib

J'ai utilisé l'interpolation spline pour lisser une série chronologique et j'aimerais également ajouter une ligne horizontale au tracé. Mais il semble y avoir un problème qui est hors de ma portée. Toute aide serait vraiment utile. Voici ce que j'ai:

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

problème semble être avec mon utilisation de [0,len(xs)]pour le traçage de lignes horizontales.

Vous avez raison, je pense que cela [0,len(xs)]vous dérange. Vous voudrez réutiliser la variable originale de l'axe des x xset la tracer avec un autre tableau numpy de la même longueur qui contient votre variable.

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

Espérons que cela résout le problème!

Vous recherchez axhline(une ligne d'axe horizontal). Par exemple, ce qui suit vous donnera une ligne horizontale à y = 0.5:

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

Si vous souhaitez dessiner une ligne horizontale dans les axes, vous pouvez également essayer la ax.hlines()méthode. Vous devez spécifier la yposition et xminet xmaxdans les coordonnées des données (c'est-à-dire votre plage de données réelle sur l'axe des x). Un exemple d'extrait de code est:

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

L'extrait ci-dessus tracera une ligne horizontale dans les axes à y=0.2. La ligne horizontale commence à x=4et se termine à x=20. L'image générée est:

Utilisez matplotlib.pyplot.hlines:

• y peut être transmis comme un emplacement unique: y=40
• y peut être transmis comme plusieurs emplacements: y=[39, 40, 41]
• Si vous tracez une figure avec quelque chose comme fig, ax = plt.subplots(), remplacez plt.hlinesou plt.axhlinepar ax.hlinesou ax.axhline, respectivement.
• matplotlib.pyplot.axhlinene peut tracer qu'un seul emplacement (par exemple 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()

En plus de la réponse la plus positive ici, on peut également enchaîner axhlineaprès avoir appelé plotun pandas's DataFrame.

import pandas as pd

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

Un moyen simple et agréable pour les personnes qui oublient toujours la commande axhlineest le suivant

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

Dans votre cas xs = xet y = 40. Si len (x) est grand, cela devient inefficace et vous devriez vraiment l'utiliser axhline.

Vous pouvez utiliser plt.gridpour dessiner une ligne horizontale.

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