用Python计算光路追迹

被几何光学里面的重复计算搞疯，所以打算拿Python当个计算器

#结构参数：
n = eval(input("折射率n："))
n0 = eval(input("折射率n'："))
r = eval(input("镜片曲率半径："))
d = eval(input("与下一面距离："))
#物方参数：
l = eval(input("物距（平行光轴入射输入值0）："))
h = eval(input("像高:"))
#判断是否平行光轴入射
if l == 0:
    i = h / r
    u = 0
    print(format(i,".4f"))
else:
    u = eval(input("输入物方孔径角u："))
    i = (l - r) / r * u
    print("i = ")
i0 = n / n0 * i
u0 = u + i - i0
l0 = r * (1 + i0 / u0)
print(format(i0,".4f"))
print(format(u0,".4f"))
print(format(l0,".4f"))

这个以后有空应该还会再改叭，还需要实现的功能是输出为表格（好像是我一直没用用熟悉的输出重定向）和实现多个面的循环
不是计科相关专业的学生，技术生疏，请轻喷，以及欢迎大家一起交流鸭