在matplotlib子图中添加一行

时间:2017-03-01 14:34:22

标签: python-3.x pandas matplotlib legend subplot

我想用matplotlib做两个数字的子图,并在两者中添加一条水平线。这可能是基本的,但我不知道如何指定应该在第一个图中绘制其中一条线,它们最终都在最后一条线中。例如

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

s1= pd.Series(np.random.rand(10))
s2= pd.Series(np.random.rand(10))

fig, axes = plt.subplots(nrows=2,ncols=1)

f1= s1.plot(ax=axes[0])
l1=plt.axhline(0.5,color='black',ls='--')
l1.set_label('l1')

f2= s1.plot(ax=axes[1])
l2=plt.axhline(0.7,color='red',ls='--') 
l2.set_label('l2')

plt.legend()

subplot with horizontal lines

axhline没有" ax"作为一个论点,就像熊猫情节函数那样。所以这会奏效:

l1=plt.axhline(0.5,color='black',ls='--',ax=axes[0])

我在matplotlib中读到了the examples,我尝试了另外一个不起作用的选项(可能是有充分理由的)

axes[0].plt.axhline(0.5,color='black',ls='--')

如何在子图中绘制线条?理想情况下有一个传奇谢谢!

1 个答案:

答案 0 :(得分:2)

在@Nick Becker的帮助下,我回答了自己的“语法”问题。

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline


s1= pd.Series(np.random.rand(10))
s2= pd.Series(np.random.randn(10))

fig, axes = plt.subplots(nrows=2,ncols=1)

f1= s1.plot(ax=axes[0],label='s1')
l1=axes[0].axhline(0.5,color='black',ls='--')
l1.set_label('l1')

axes[0].legend(loc='best')

f2= s1.plot(ax=axes[1],label='s2')

l2=axes[1].axhline(0.5,color='black',ls='--')

l2.set_label('l2')

axes[1].legend(loc='best')

subplots with horizontal lines