arq*_*ago 4 python normal-distribution matplotlib
我希望在 python 中创建一个标准的正态分布(平均值 = 0,标准偏差 = 1)曲线,然后在 z 分数的左侧、右侧和中间创建阴影区域。我还想打印 z-score(s) 和阴影区域的相关概率。
例如,我感兴趣的阴影区域是:
概率(z < -0.75)
概率(z > 0.75)
概率(-0.75 < z < 0.75)
我使用以下几行来创建标准正态分布曲线。如何为关联的 z 分数添加阴影区域并将 z 分数与概率一起打印?
import numpy as np
import matplotlib.pyplot as plt
from scipy.stats import norm
range = np.arange(-3,3,0.001)
plt.plot(range, norm.pdf(range, 0, 1))
plt.show()
Run Code Online (Sandbox Code Playgroud)
您必须使用fill_between绘制 2 条曲线之间区域的函数,在这种情况下,在y = 0和之间y = normal distribution,以方便任务已创建以下函数:
def draw_z_score(x, cond, mu, sigma, title):
y = norm.pdf(x, mu, sigma)
z = x[cond]
plt.plot(x, y)
plt.fill_between(z, 0, norm.pdf(z, mu, sigma))
plt.title(title)
plt.show()
Run Code Online (Sandbox Code Playgroud)
例子:
x = np.arange(-3,3,0.001)
z0 = -0.75
draw_z_score(x, x<z0, 0, 1, 'z<-0.75')
Run Code Online (Sandbox Code Playgroud)
输出:
x = np.arange(-3,3,0.001)
z0 = 0.75
draw_z_score(x, (-z0 < x) & (x < z0), 0, 1, '-0.75<z<0.75')
Run Code Online (Sandbox Code Playgroud)
输出:
x = np.arange(-3,3,0.001)
z0 = 0.75
draw_z_score(x, x > z0, 0, 1, ' z> 0.75')
Run Code Online (Sandbox Code Playgroud)
输出: