基于数据样本的正态分布图实现

#from mpl_toolkits.mplot3d.axes3d import Axes3D
#import matplotlib.pyplot as plt
from scipy.stats import norm
import scipy.stats as stats
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
plt.rcParams["figure.figsize"]=(14,7)

plt.figure(dpi=100)

mu_real=10
sigma_real=2

np.random.seed(42)
sample=stats.norm.rvs(loc=mu_real,scale=sigma_real,size=1000)

mu_est=np.mean(sample)
sigma_est=np.std(sample)
print("Estimated MU:{}\nEstimated SIGMA:{}".format(mu_est,sigma_est))

plt.hist(sample,bins=50,normed=True,alpha=.25)

plt.plot(np.linspace(2,18,1000),norm.pdf(np.linspace(2,19,1000),\
                                         loc=mu_real,scale=sigma_real),color="red",linestyle="dashed")

plt.plot(np.linspace(2,18,1000),norm.pdf(np.linspace(2,19,1000),\
                                         loc=mu_mean(sample),scale=np.std(sample)),color="green",linewidth=2)

plt.text(x=9.5,y=.1,s="sample",alpha=.75,weight="bold",color="#008fd5")
plt.text(x=7,y=.2,s="estimated distribution",rotation=55,alpha=.75,weight="bold",\
         color="green")

plt.tick_params(axis='both',which='major',labelsize=18)
plt.axhline(y=0,color="black",linewidth=1.3,alpha=.7)

plt.text(x=0,y=0.3,s="Normal Distribution",fontsize=26,weight="bold",alpha=.75)