import numpy as np
import math
import matplotlib.pyplot as plt

n = 25 # количество элементов АФАР
wavelength = (3e8)/(2.4e9) # длина волны для несущей частоты 2.4 ГГц
d = wavelength / 2 
k = 2*np.pi/wavelength
thetaM = 0*(np.pi/180)

thetas = np.arange(-90,91)*(np.pi/180)
for idx, item in enumerate(thetas):
    if item == 0:
        thetas[idx] = 0.001
        
F = np.zeros((len(thetas),1))
for idx, theta in enumerate(thetas):
    num = np.sin(n*k*d*(np.sin(theta) - np.sin(thetaM))/2)
    denum = np.sin(k*d*(np.sin(theta) - np.sin(thetaM))/2)
    F[idx] = (1 /n)*(num/denum)
    if math.isnan(F[idx]) == True:
        F[idx] = 1

plt.figure(figsize=(10, 5), dpi=150)
plt.plot(thetas*(180/np.pi), F)
plt.title("Азимут")
plt.xlabel('Градусы')
plt.ylabel('F(θ)')
plt.grid()
plt.savefig('LA')