from scipy.stats import norm
from math import sqrt
from math import ceil

s = ['karavan', 'karavan', 'limuzina', 'nimam avtomobila', 'limuzina', 'karavan', 'nimam avtomobila', 'karavan', 'karavan', 'nimam avtomobila', 'nimam avtomobila', 'kupe', 'nimam avtomobila', 'kabriolet', 'kupe', 'limuzina', 'enoprostorec', 'kabriolet', 'limuzina', 'kabriolet', 'limuzina', 'kabriolet', 'karavan', 'kupe', 'kabriolet', 'kupe', 'enoprostorec', 'kupe', 'kupe', 'karavan', 'kupe', 'karavan', 'nimam avtomobila', 'limuzina', 'kabriolet', 'limuzina', 'karavan', 'karavan', 'enoprostorec', 'karavan', 'enoprostorec', 'limuzina', 'kabriolet', 'enoprostorec', 'kupe', 'nimam avtomobila', 'kupe', 'limuzina', 'limuzina', 'enoprostorec', 'kabriolet', 'nimam avtomobila', 'kupe', 'kabriolet', 'enoprostorec', 'limuzina', 'enoprostorec', 'karavan', 'enoprostorec', 'kabriolet', 'nimam avtomobila', 'kabriolet', 'nimam avtomobila', 'limuzina', 'kabriolet', 'nimam avtomobila', 'limuzina', 'nimam avtomobila', 'kupe', 'karavan', 'kabriolet', 'nimam avtomobila', 'nimam avtomobila', 'kupe', 'karavan', 'karavan', 'kabriolet', 'enoprostorec', 'kabriolet', 'kabriolet', 'nimam avtomobila', 'kabriolet', 'limuzina', 'limuzina', 'karavan', 'kupe', 'kabriolet', 'limuzina', 'kabriolet', 'nimam avtomobila', 'nimam avtomobila', 'nimam avtomobila', 'kabriolet', 'enoprostorec', 'limuzina', 'limuzina', 'kupe']

n = len(s)
iskanDelez = 'kupe'
st = s.count(iskanDelez)
delez = st / len(s)

print("Delez anketirancev: " + str(delez))

k=st

delezVzorca=delez

beta=0.95

z=norm.isf(1-(1+beta)/2)

Delta=z*sqrt(delezVzorca*(1-delezVzorca)/n)

spodnjaMeja=delezVzorca-Delta
zgornjaMeja=delezVzorca+Delta

print("Interval zaupanja pri stopnji zaupanja {0} je [{1},{2}].".format(beta,spodnjaMeja,zgornjaMeja))

delezVzorcaNaloge = 0.18
intervalKrajsiOd = 0.04
print("Velikost vzorca je {0}.".format(
        ceil(1/((intervalKrajsiOd/2/z)**2/delezVzorcaNaloge/(1-delezVzorcaNaloge)))))