import numpy as np # 导入 numpy 库
import matplotlib.pyplot as plt # 导入 matplotlib.pyplot 库
from scipy.stats import binom # 从 scipy.stats 导入 binom 模块

设置试验次数和成功概率

n = 6
p = 0.3

使用 numpy 的 arange 函数生成从 0 到 6 的数组

x = np.arange(7)

计算二项分布的概率质量函数

y = binom.pmf(x, n, p)

创建 1 行 2 列的子图

plt.subplot(121)
plt.plot(x, y, 'ro') # 绘制点
plt.vlines(x, 0, y, 'k', lw=2, alpha=0.5) # 绘制垂直线
plt.title('Vertical Lines Plot of Binomial PMF') # 添加标题

创建第二个子图

plt.subplot(122)
plt.stem(x, y) # 绘制茎图
plt.title('Stem Plot of Binomial PMF') # 添加标题

调整布局以防止子图重叠（可选）

plt.tight_layout()

保存图形到文件

plt.savefig("figure9_2.png", dpi=500)

显示图形

plt.show()

print("学号后两位：04")