PTA 7-12 查询自动售货机中商品的价格 (Python)

PS:选择语句的顺序真的很重要

import sys
print("[1] crisps")
print("[2] popcorn")
print("[3] chocolate")
print("[4] cola")
print("[0] exit")
count=0
dit={'1':3.0,'2':2.5,'3':4.0,'4':3.5,'0':0}
N=list(input().split(' '))
for i in N:
    count+=1
    if int(i)<0 or int(i)>4:
        print("price = 0.0")
    elif dit[i]!=0:
        print(f"price = {dit[i]}")
        if count==5:
            print("Thanks")
            sys.exit()
    else:
        print("Thanks")
        sys.exit()