Лабораторная 2 / Вариант 4 / Lab2

import matplotlib.pyplot as plt
import numpy as np
import random
import time


class RandomFibon:
    def __init__(self, k, m, debug=False):
        self.k = k
        self.m = m
        if debug:
            random.seed(0)
            self.A = [random.randint(0, 2**m) for _ in range(17)]
        else:
            random.seed(time.time())
            self.A = [random.randint(0, 2**m) for _ in range(17)]

        if (k > 16 or m % 2 == 1):
            raise Exception

    def next(self):
        x = (self.A[-1] + self.A[-self.k-1]) % 2 ** self.m
        self.A.pop(0)
        self.A.append(x)

        return x / 2 ** self.m


class RandomValue:
    def __init__(self, xj, pj):
        self.Pj = [sum(pj[:i+1]) for i in range(len(pj))]
        self.xj = xj
        self.pj = pj
        self.z = RandomFibon(10, 32, debug=True)

    def next(self):
        z = self.z.next()
        for p in range(len(self.Pj)):
            if z <= self.Pj[p]:
                return self.xj[p]


x = np.array([-80.1, -77.2, -11.6, -11.3, 36.2, 42.4, 79.1])
p = [0.079, 0.122, 0.116, 0.205, 0.164, 0.158, 0.156]
rv = RandomValue(x, p)
RV = []
N = 500
for i in range(N):
    RV.append(rv.next())

print("Первые 30 значений:")
for i in range(30):
    print(RV[i])
M = sum(RV) / N
D = sum([x ** 2 for x in RV]) / N - M ** 2
print(f"Эмпирические: M = {M:.5f}, D = {D:.5f}")

M_exp = sum([p[j] * x[j] for j in range(len(p))])
D_exp = sum([p[j] * x[j] ** 2 for j in range(len(p))]) - M_exp ** 2
print(f"Теоретические: M = {M_exp:.5f}, D = {D_exp:.5f}")


edges = (x[:-1] + x[1:]) / 2
bins = np.concatenate(
    ([x[0] - (edges[0] - x[0])], edges, [x[-1] + (x[-1] - edges[-1])]))

plt.hist([RV, x], bins=bins,
         weights=[np.ones_like(RV) / len(RV), np.ones_like(x) * p],
         label=['Эмпирическое', 'Теоретическое'])
plt.legend(loc='upper left')
plt.savefig("Sonya/ms2.svg")
plt.show()