用置信区间python绘制优势比图

import numpy as np
import statsmodels.api as sm
import matplotlib.pyplot as plt

# 生成两组随机数据
group1 = np.random.binomial(1, 0.6, size=100)  # 成功率为0.6的处理方法1
group2 = np.random.binomial(1, 0.8, size=100)  # 成功率为0.8的处理方法2

# 计算优势比及其置信区间
count1 = np.sum(group1)
count2 = np.sum(group2)
nobs1 = len(group1)
nobs2 = len(group2)

# 使用statsmodels计算优势比及其置信区间
oddsratio, ci_lower, ci_upper = sm.stats.proportion.proportion_confint([count1, count2], [nobs1, nobs2], method='normal')

# 绘制优势比图
plt.errorbar([1, 2], [count1/nobs1, count2/nobs2], yerr=[(ci_lower[0]/nobs1, ci_lower[1]/nobs2), (ci_upper[0]/nobs1, ci_upper[1]/nobs2)], fmt='o')
plt.xticks([1, 2], ['Method 1', 'Method 2'])
plt.ylabel('Proportion')
plt.title('Comparison of Proportions')
plt.show()