加法定则验证

def solve():
    import pandas as pd
    import matplotlib.pyplot as plt
    url = 'https://data.zuihe.com/titanic.csv'
    df = pd.read_csv(url)
    n = len(df)
    p_s = (df['survived']==1).sum()/n
    p_f = (df['sex']=='female').sum()/n
    p_sf = ((df['survived']==1)&(df['sex']=='female')).sum()/n
    f = round(p_s+p_f-p_sf, 4)
    d = round(((df['survived']==1)|(df['sex']=='female')).sum()/n, 4)
    print(f'加法定则={f}，直接计算={d}，验证={"通过" if f==d else "不通过"}')
    labels = ['P(生还)','P(女性)','P(生还∩女性)','P(生还∪女性)']
    values = [p_s, p_f, p_sf, d]
    plt.bar(labels, values, color='steelblue', edgecolor='black')
    plt.ylabel('概率'); plt.title('概率加法定则验证'); plt.xticks(rotation=15)
    plt.tight_layout(); plt.show()