如果使用用戶定義的函數輸入數字，如何添加數字

Question

如果使用用戶定義的函數輸入整數，是否可以添加整數？ 下面是我的示例代碼

def first_num(num1):
    num1 = int(input("Enter first number: "))
    return 

def sec_num(num2):
    num2 = int(input("Enter second number: "))
    return 

print("ARITHMETIC CALCULATOR")
print("----------------")
print("[1] - Addition")
print("[2] - Subtraction")
print("[3] - Multiplication")
print("[4] - Division")
print("----------------")

choice = int(input("Enter your choice: "))
first_num(num1)
sec_num(num2)

if choice == '1':
    sum = first_num(num1) + sec_num(num2)
    print("The sum is: ")

我只是想問一下是否可以，如果可以，如何？

Answer 1

您的函數需要return您要求的數字，並且您需要在調用它時將結果分配給一個變量。 由於您可以多次調用函數，因此無需為每個數字定義唯一的函數。

def get_num(label):
    return int(input(f"Enter {label} number: "))


print("ARITHMETIC CALCULATOR")
print("----------------")
print("[1] - Addition")
print("[2] - Subtraction")
print("[3] - Multiplication")
print("[4] - Division")
print("----------------")

choice = input("Enter your choice: ")
num1 = get_num("first")
num2 = get_num("second")

if choice == '1':
    print(f"The sum is: {num1 + num2}")

ARITHMETIC CALCULATOR
----------------
[1] - Addition
[2] - Subtraction
[3] - Multiplication
[4] - Division
----------------
Enter your choice: 1
Enter first number: 31
Enter second number: 11
The sum is: 42