[Python] 간단한 계산기

파이썬 기본 문법에 익숙해지기 위해 이것 저것 개발해보고 있다. 이번에는 간단한 계산기를 개발해보았다.

너무 간단하지만 기본 문법이 익숙해져야 하니..

def add(a, b):
    return a + b

def subtract(a, b):
    return a - b

def multiply(a, b):
    return a * b

def divide(a, b):
    return a / b

print("Simple Calculator")

while True:
    print("0. Quit")
    print("1. Addition")
    print("2. Subtraction")
    print("3. Multiplication")
    print("4. Division")

    choice = input("Enter your choice (0-4): ")

    if choice == '0':
        print("Bye")
        break

    num1 = float(input("Enter first number: "))
    num2 = float(input("Enter second number: "))

    if choice == '1':
        print(num1, "+", num2, "=", add(num1, num2))
    elif choice == '2':
        print(num1, "-", num2, "=", subtract(num1, num2))
    elif choice == '3':
        print(num1, "*", num2, "=", multiply(num1, num2))
    elif choice == '4':
        print(num1, "/", num2, "=", divide(num1, num2))
    else:
        print("Invalid input")