二项分布概率计算

def solve():
    import matplotlib.pyplot as plt
    import numpy as np
    from scipy import stats
    n, p = 10, 0.5
    prob_6 = stats.binom.pmf(6, n, p)
    print(f"恰好6次正面的概率: {prob_6:.4f}")
    x = np.arange(0, n + 1)
    pmf = stats.binom.pmf(x, n, p)
    plt.figure(figsize=(10, 6))
    plt.bar(x, pmf, edgecolor='black', alpha=0.7)
    plt.bar(6, prob_6, color='red', edgecolor='black', alpha=0.7, label=f'P(X=6)={prob_6:.4f}')
    plt.xlabel('正面次数')
    plt.ylabel('概率')
    plt.title('二项分布 B(10, 0.5)')
    plt.legend()
    plt.grid(alpha=0.3, axis='y')
    plt.show()