# standard error for proportion
# p hat: estimate proportion for population
p = 0.6
n = 100
std_err_prop = math.sqrt(p * (1 - p) / n) * 100
print "Sai số ước lượng:", std_err_prop
print "Tỉ lệ chặn dưới:", p * 100 - std_err_prop
print "Tỉ lệ chặn trên:", p * 100 + std_err_prop

# increase sample size will decrease standard error
n = 1000
std_err_prop = math.sqrt(p * (1 - p) / n) * 100
print "Sai số ước lượng:", std_err_prop
print "Tỉ lệ chặn dưới:", p * 100 - std_err_prop
print "Tỉ lệ chặn trên:", p * 100 + std_err_prop

# lấy mẫu 5 khách hàng vào mỗi ngày thứ Hai như sau
n = 5
sample_A = np.asarray([2.1, 1.3, 0.6, 1.5, 2.4])
sample_B = np.asarray([0.7, 1.3, 1.8, 0.7, 1.9])
sample_C = np.asarray([2.1, 1.9, 2.5, 1.8, 0.9])
sample_D = np.asarray([1.6, 2.1, 1, 0.9, 1.2])
print "Sample A mean:", sample_A.mean()
print "Sample B mean:", sample_B.mean()
print "Sample C mean:", sample_C.mean()
print "Sample D mean:", sample_D.mean()

average_of_means = (sample_A.mean() + sample_B.mean() + sample_C.mean() + sample_D.mean()) / 4
list_of_means = np.asarray([sample_A.mean(), sample_B.mean(), sample_C.mean(), sample_D.mean()])
std_of_means = list_of_means.std()
print "Thời gian trung bình từ các mẫu trung bình:", average_of_means
print "Độ lệch của từ các mẫu trung bình:", std_of_means

# estimate standard deviation of population
std_of_population = std_of_means * math.sqrt(n)
print "Ước lượng độ lệch chuẩn cho quần thể:", std_of_population
print "Chặn dưới cho ước lượng:", average_of_means - std_of_means
print "Chặn trên cho ước lượng:", average_of_means + std_of_means

# lấy mẫu 100 người
n = 100

# ủng hộ ứng viên A là 55%
candidate_A_prop = 0.55

# ủng hộ ứng viên B là 45%
candidate_B_prop = 0.45

# tính Zscore cho ứng viên A với độ tin cậy là 95%
qnorm = scipy.stats.norm.ppf(0.975)
print "Z-score:", qnorm

# calculate p hat
p = candidate_A_prop
std_err_prop = math.sqrt(p * (1 - p) / n)
print "Độ lỗi chuẩn cho tỉ lệ ủng hộ ứng viên A:", std_err_prop

# calculate standard error of mean
std_of_means = std_err_prop / math.sqrt(n)
print "Độ lỗi chuẩn cho tỉ lệ thu thập được:", std_of_means

# calculate interval
upper_limit = candidate_A_prop + qnorm * std_of_means
lower_limit = candidate_A_prop - qnorm * std_of_means
print "Tỉ lệ ủng hộ chặn trên:", upper_limit
print "Tỉ lệ ủng hộ chặn dưới:", lower_limit

# thống kê thời gian các lượt chạy
list_run_time = np.asarray([11.3, 10.3, 9.6, 9.1, 8.9, 8.9, 8.1, 7.5, 6.9])
sample_size = list_run_time.size
sample_mean = list_run_time.mean()
sample_std = list_run_time.std()
standard_err = sample_std / math.sqrt(sample_size)

print "Total runners:", sample_size
print "Sample mean:", sample_mean
print "Sample standard deviation:", sample_std
print "Standard/Sampling error:", standard_err

# calculate interval
qnorm = scipy.stats.norm.ppf(0.975)
print "Z-score:", qnorm
upper_limit = sample_mean + qnorm * standard_err
lower_limit = sample_mean - qnorm * standard_err
print "Thời gian trung bình chặn trên:", upper_limit
print "Thời gian trung bình chặn dưới:", lower_limit

n = n_boys + n_girls
x = n_girls
p = 0.51
alpha = 0.05
p_value = scipy.stats.binom.pmf(x, n, p)
print "p-value/Z-score:", p_value

# reject H0 or not
if p_value < alpha:
    print "Reject H0 => nữ bị gọi là do thiên vị"
else:
    print "Accept H0 => nữ bị gọi là do tình cờ"

# lấy mẫu thực nghiệm
n = 500
p_hat = yes = 0.54
p_0 = 0.5
z_p = (p_hat - p_0)/(math.sqrt(p_0 * (1 - p_0) / n))
print "Z-score:", z_p

# one-tail test
z_cdf = scipy.stats.norm.cdf(z_p)
print "Z-cdf:", z_cdf

# calculate p-value
p_value = 1 - z_cdf
print "p-value:", p_value

# reject H0 or not
if p_value < alpha:
    print "Reject H0 => ứng viên thắng cử"
else:
    print "Accept H0 => ứng viên thua cử"

# lấy mẫu thực nghiệm
n = 100
mu = 20.15
avg_weight = 20.1
std_weight = 0.26
z = (avg_weight - mu)/(std_weight / math.sqrt(n))
print "Z-score:", z

# one-tail test
p_value = scipy.stats.norm.cdf(z)
print "p-value:", p_value

# reject H0 or not
if p_value < alpha:
    print "Reject H0 => không ship hàng"
else:
    print "Accept H0 => ship hàng"