我想做一个简单的计算器使用python和一个for循环。我有办法让它工作吗？

def calculator():
    while True:
        num1 = float(input("请输入第一个数字："))
        operator = input("请输入运算符（+、-、*、/）：")
        num2 = float(input("请输入第二个数字："))

        if operator == '+':
            result = num1 + num2
        elif operator == '-':
            result = num1 - num2
        elif operator == '*':
            result = num1 * num2
        elif operator == '/':
            result = num1 / num2
        else:
            print("无效的运算符！")
            continue

        print("结果：", result)

        choice = input("是否继续计算？（y/n）")
        if choice.lower() != 'y':
            break

calculator()