Matematyka_Bayes/main.py
2021-05-31 19:11:26 +02:00

93 lines
2.6 KiB
Python

import pandas as pd
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("ssssss",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(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(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(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(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(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(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(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(p_sugar_if_heart_attack)
licznik=p_man_if_heart_attack*p_sugar_if_heart_attack*p_over50_if_heart_attack*p_chol_over200_if_heart_attack*p_heart_attack
mianownik=(p_man_if_heart_attack*p_sugar_if_heart_attack*p_over50_if_heart_attack*p_chol_over200_if_heart_attack*p_heart_attack)+((1-p_man_if_heart_attack)*(1-p_sugar_if_heart_attack)*(1-p_over50_if_heart_attack)*(1-p_chol_over200_if_heart_attack)*(1-p_heart_attack))
print(licznik)
print(mianownik)
print(licznik/mianownik)