lab1_1_add_task.py

import random
import math
import matplotlib.pyplot as plt

n = 10
omegaMax = 1200
N = 64

x = []
y = []

a = []
b = []

Dx = 0
Mx = 0

def Plot():
    A = []
    fi = []
    for i in range(n):
        A.append(random.random())
        fii = random.random() * omegaMax
        fi.append(fii)
    for i in range(N):
        res = 0
        for j in range(n):
            res += A[j] * math.sin((omegaMax / (j + 1)) * i + fi[j])
        x.append(res)
        yy = i
        y.append(yy)

def Expectancy():
    Mxx = 0
    for t in range(N):
        Mxx += (1 / N) * x[t]
    return Mxx

def Dispersion():
    Dx = 0
    for t in range(N):
        Dx += (1 / (N - 1)) * (x[t] - Mx) * (x[t] - Mx)
    return Dx

#additional task
def Plott():
    Dx = 0
    for l in range(N):
        aa = l
        a.append(aa)
    for t in range(N):
        Dx += (1 / (N - 1)) * (x[t] - Mx) * (x[t] - Mx)
        bb = Dx
        b.append(bb)

if __name__ == "__main__":
    Plot()
    Plott()
    print(Expectancy())
    print(Dispersion())
    #plt.plot(y, x)
    plt.ylabel('N')
    plt.xlabel('Dispersion')
    plt.plot(b, a)
    plt.show()