在pycharm中绘制正太分布图

1.目标

在pycharm中绘制正太分布图

2.代码

# @Project: zc
# @Author: zc
# @File name: distribution
# @Create time: 2021/11/15 22:10
# 描述：在pycharm中绘制正太分布图

# 1.导入相关库
import numpy as np
from d2l import torch as d2l
import matplotlib.pyplot as plt


# 2.定义正太分布密度函数
def normal(x, mu, sigma):
	p = 1.0 / (np.sqrt(2 * np.pi * sigma ** 2))
	return p * np.exp(-0.5 / (sigma ** 2) * (x - mu) ** 2)


# 3.生成 x,y值
x = np.arange(-7, 7, 0.01)  # 从 -7 到 7，以 0.01 为间隔

# 4.定义多组参数 mu,sigma
params = [(0, 1), (0, 2), (3, 1)]

# 5.绘制正太分布图
d2l.plot(x, [normal(x, mu, sigma) for mu, sigma in params], xlabel='x', ylabel='p(x)',
		 legend=[f'mean={mu},std={sigma}' for mu, sigma in params])

# 6.显示图片
plt.show()

3.结果