"""
@Project ：microproduct 
@File ：OnePlantHeight.PY
@Function ：主函数
@Author ：LMM
@Date ：2021/10/19 14:39
@Version ：1.0.0
"""

import numpy as np
from scipy.special import lambertw
H_arr=np.array([1,2])
array_angle=np.array([1,0.5])
a = 0.5 * H_arr * H_arr
b = -0.5 * np.sin(array_angle)
y = 2



a1 = 2 * lambertw(-b * np.sqrt(y / a) / 2) / b
print(a1)
pass