[파이썬] 문제 : 두 수 입력받아 사칙연산 함수 만들기

 

def add(a, b):     # 더하기 함수
    return a + b

def sub(a, b):     # 빼기 함수
    return a - b
 
def mul(a, b):     # 곱하기 함수
    return a*b

def div(a, b):     # 나누기 함수
    return a/b

a1 = int(input('정수1 : '))     # 정수 1 입력 
op = input('연산자(+,-,*,/) :') # 연산자 입력
b1 = int(input('정수2 : '))     # 정수 2 입력

if op == '+':             # 연산자가 + 기면 더하기 함수 실행
    print(add(a1,  b1))
elif op == '-':           # 연산자가 - 기면 빼기 함수 실행
    print(sub(a1, b1))
elif op == '*':           # 연산자가 * 기면 곱하기 함수 실행
    print(mul(a1, b1))
elif op == '/':           # 연산자가 / 기면 나누기 함수 실행
    print(div(a1, b1))

결과

정수1 : 4
연산자(+,-,*,/) :-
정수2 : 2
2