92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
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)
|