Untitled

from math import tan
from numpy import radians

variant = 11 # Номер варианта

dm = 1.4
P = 0
if (variant > 0) and (variant < 6):
    P += 1
elif (variant >= 6) and (variant < 11):
    P += 2
elif (variant >= 11) and (variant < 16):
    P += 3
elif (variant >= 16) and (variant < 20):
    P += 4

M_inf0 = 2
H0 = 3.5
bettak0 = 7.5
x_cm = 0.65

bettak = bettak0 + 2.5 * (variant - 2 * P)  # Угол конуса
lk = dm / (2 * tan(radians(bettak)))        # Длина конуса
Xcm = x_cm * lk                             # Координата центра масс
l_s_chertoy_k = (2 * lk) / 3                # Длина затупленного конуса

# Данные для атмосферы
h = (H0 + 0.2 * variant) * 1000             # Геометрическая высота
M_inf = M_inf0 + (P - 1)                    # Мах
r = 6356767
g0 = 9.80665
R = 287.05287