动手学深度学习-d2l的plot使用

绘制函数y=x的3次方-x的-1次方在x=1处的切线

x = np.arange(0, 3, 0.1)
d2l.plot(x, [f(x), 4*(x-1)], 'x', 'f(x)', legend=['f(x)', 'Tangent line (x=1)'])

绘制不同均值、不同方差的正态分布

def normal(x,mu,sigma):
    p=1/math.sqrt(2*math.pi*sigma**2)
    return p*np.exp(-0.5/sigma**2*(x-mu)**2)
x=np.arange(-7,7,0.01)
params=[(0,1),(0,2),(3,1)]
d2l.plot(x,[normal(x,mu,sigma) for mu,sigma in params],xlabel='x',ylabel='p(x)',figsize=(4.5,2.5),legend=[f'mean:{mu},std:{sigma}'for mu,sigma in params])