Chương trình máy tính cơ bản trong Python

Xây dựng chương trình máy tính đơn giản thực hiện 4 phép toán: +, -, *, /.

Code hoàn chỉnh

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):
    if b == 0:
        return "Lỗi: Không chia cho 0"
    return a / b

print("Chọn phép toán:")
print("+ : Cộng")
print("- : Trừ")
print("* : Nhân")
print("/ : Chia")

choice = input("Nhập phép toán (+, -, *, /): ")
a = float(input("Nhập số thứ nhất: "))
b = float(input("Nhập số thứ hai: "))

if choice == '+':
    print(f"{a} + {b} = {add(a, b)}")
elif choice == '-':
    print(f"{a} - {b} = {subtract(a, b)}")
elif choice == '*':
    print(f"{a} * {b} = {multiply(a, b)}")
elif choice == '/':
    result = divide(a, b)
    print(f"{a} / {b} = {result}")
else:
    print("Phép toán không hợp lệ")

Kết quả mẫu

Chọn phép toán:
+ : Cộng
- : Trừ
* : Nhân
/ : Chia
Nhập phép toán (+, -, *, /): -
Nhập số thứ nhất: 34
Nhập số thứ hai: 20
34.0 - 20.0 = 14.0

Cải tiến gợi ý

• Thêm xử lý ngoại lệ try/except cho input không phải số
• Vòng lặp cho phép tính toán liên tục
• Thêm các phép toán: lũy thừa (**), chia lấy dư (%)
• Dùng dictionary mapping operator → function để code clean hơn

Phiên bản nâng cao (dictionary mapping)

operations = {
    '+': lambda a, b: a + b,
    '-': lambda a, b: a - b,
    '*': lambda a, b: a * b,
    '/': lambda a, b: a / b if b != 0 else "Lỗi: Chia cho 0"
}

op = input("Phép toán: ")
a, b = float(input("a: ")), float(input("b: "))

if op in operations:
    print(f"Kết quả: {operations[op](a, b)}")
else:
    print("Phép toán không hợp lệ")