from Atom import Atom

for i in range(18, 24):  # Ar - V
    a = Atom(i, q=0)  # 原子序数为i,带电量为q的微粒
    print("Atom: " + str(a.Index))
    print("    3p:" + str(a.getEnergy(3, 'p')))
    print("    3d:" + str(a.getEnergy(3, 'd')))
    print("    4s:" + str(a.getEnergy(4, 's')))

print('\n\n\n Finding 4s < 3d ...')  # 尝试寻找4s能量<3d能量的轨道……
for i in range(19, 25):
    a = Atom(i)
    if (a.getEnergy(3, 'd') > a.getEnergy(4, 's')):
        print(a.Index)