import pandas as pd


#       P(B|A)*P(A)
#P(A|B)=-----------
#          P(B)


data=pd.read_csv('heart.csv')

print(data)

men=0
for i in range(len(data['sex'])):
    if data['sex'][i] ==1:
        men+=1
print(men)
p_men=men/len(data)
print("Prawdopodobieństwo, że mężczyzna",p_men)

chol=0
for i in range(len(data['chol'])):
    if data['chol'][i] >200:
        chol+=1
print(chol)
p_chol=chol/len(data)
print("Prawdopodobieństwo, że cholesterol większy niż 200",p_chol)

age_over_50=0
for i in range(len(data['age'])):
    if data['age'][i] >50:
        age_over_50+=1
print(age_over_50)
p_age_over_50=age_over_50/len(data)
print("Prawdopodobieństwo, że wiek powyżej 50",p_age_over_50)

sugar=0
for i in range(len(data['fbs'])):
    if data['fbs'][i] ==1:
        sugar+=1
print(sugar)
p_sugar=sugar/len(data)
print("Prawdopodobieństwo, że wysoki cukier",p_sugar)

heart_attack=0
for i in range(len(data['target'])):
    if data['target'][i] ==1:
        heart_attack+=1
print(heart_attack)
p_heart_attack=heart_attack/len(data)
print("Prawdopodobieństwo dużego ryzyka zawału serca",p_heart_attack)   #P(class)

man_if_heart_attack=0
for i in range(len(data['target'])):
    if data['target'][i] ==1:
        if data['sex'][i]==1:
            man_if_heart_attack+=1
print(man_if_heart_attack)
p_man_if_heart_attack=man_if_heart_attack/len(data)
p_man_if_heart_attack=p_man_if_heart_attack/p_heart_attack
print("rawdopodobieństwo, że mężczyzna jeśli wystąpił zawał serca",p_man_if_heart_attack)

over50_if_heart_attack=0
for i in range(len(data['target'])):
    if data['target'][i] ==1:
        if data['age'][i]>50:
            over50_if_heart_attack+=1
print(over50_if_heart_attack)
p_over50_if_heart_attack=over50_if_heart_attack/len(data)
p_over50_if_heart_attack=p_over50_if_heart_attack/p_heart_attack
print("Prawdopodobieństwo, że powyżej 50 lat jeżeli wystąpił zawał serca",p_over50_if_heart_attack)

chol_over200_if_heart_attack=0
for i in range(len(data['target'])):
    if data['target'][i] ==1:
        if data['chol'][i]>200:
            chol_over200_if_heart_attack+=1
print(chol_over200_if_heart_attack)
p_chol_over200_if_heart_attack=chol_over200_if_heart_attack/len(data)
p_chol_over200_if_heart_attack=p_chol_over200_if_heart_attack/p_heart_attack
print("Prawdopodobieństwo, że cholestorol powyżej 200, jezeli wystąpił zawał serca",p_chol_over200_if_heart_attack)

sugar_if_heart_attack=0
for i in range(len(data['target'])):
    if data['target'][i] ==1:
        if data['fbs'][i]==1:
            sugar_if_heart_attack+=1
print(sugar_if_heart_attack)
p_sugar_if_heart_attack=sugar_if_heart_attack/len(data)
p_sugar_if_heart_attack=p_sugar_if_heart_attack/p_heart_attack
print("Prawdopodbieństwo, że był wysoki poziom cukru jeżeli wystąpił zawał serca",p_sugar_if_heart_attack)