단순 사칙연산 계산기 파이썬 코드

단순 사칙연산을 수행하는 계산기 파이썬 코드입니다.

def add(x, y):
    return x + y

def subtract(x, y):
    return x - y

def multiply(x, y):
    return x * y

def divide(x, y):
    if y == 0:
        return "'0'으로 나눌 수 없습니다. 다시 시도해주세요."
    return x / y

def calculator():
    print("다음 중 골라주세요:")
    print("1. 덧셈")
    print("2. 뺄셈")
    print("3. 곱셈")
    print("4. 나눗셈")

    while True:
        choice = input("더하기 1, 빼기 2, 곱하기 3, 나누기 4 중 선택해주세요: ")

        if choice in ['1', '2', '3', '4']:
            try:
                num1 = float(input("계산을 위한 첫 번째 숫자를 입력해주세요: "))
                num2 = float(input("계산을 위한 두 번째 숫자를 입력해주세요: "))
            except ValueError:
                print("잘못 입력하였습니다. 숫자를 입력해주세요.")
                continue

            if choice == '1':
                print(f"{num1} + {num2} = {add(num1, num2)}")
            elif choice == '2':
                print(f"{num1} - {num2} = {subtract(num1, num2)}")
            elif choice == '3':
                print(f"{num1} * {num2} = {multiply(num1, num2)}")
            elif choice == '4':
                print(f"{num1} / {num2} = {divide(num1, num2)}")
        else:
            print("잘못 입력하였습니다. 1~4 중에서 선택해주세요.")
        
        next_calculation = input("이어서 다른 계산을 할 것인가요? (네/아니오)")
        if next_calculation.lower() != '네':
            break

if __name__ == "__main__":
    calculator()