diff --git a/solution.py b/solution.py index 491e11d..df2e1e2 100644 --- a/solution.py +++ b/solution.py @@ -1,17 +1,22 @@ import pandas as pd import numpy as np - from scipy.stats import norm +from scipy.stats import chi2 - -chat_id = 123456 # Ваш chat ID, не меняйте название переменной +chat_id = 402739329 # Ваш chat ID, не меняйте название переменной def solution(p: float, x: np.array) -> tuple: # Измените код этой функции # Это будет вашим решением # Не меняйте название функции и её аргументы alpha = 1 - p - loc = x.mean() - scale = np.sqrt(np.var(x)) / np.sqrt(len(x)) - return loc - scale * norm.ppf(1 - alpha / 2), \ - loc - scale * norm.ppf(alpha / 2) + n = len(x) + x2 = np.array([xi**2 for xi in x]) + x2_mean = x2.mean() + + chi2_rv = chi2(df = 2 * n) + + left = chi2_rv.ppf(1 - alpha / 2) + right = chi2_rv.ppf(alpha / 2) + + return np.sqrt(n * x2_mean / (left * 23)), np.sqrt(n * x2_mean / (right * 23))