python中问题的自动生成

import random

def generate_question():
    # 生成两个随机数
    num1 = random.randint(1, 10)
    num2 = random.randint(1, 10)
    
    # 随机选择一个运算符
    operator = random.choice(['+', '-', '*', '/'])
    
    # 根据运算符生成问题和答案
    if operator == '+':
        question = f"What is {num1} + {num2}?"
        answer = num1 + num2
    elif operator == '-':
        question = f"What is {num1} - {num2}?"
        answer = num1 - num2
    elif operator == '*':
        question = f"What is {num1} * {num2}?"
        answer = num1 * num2
    else:
        # 避免除法出现小数
        num1 = num2 * random.randint(1, 10)
        question = f"What is {num1} / {num2}?"
        answer = num1 / num2
    
    return question, answer

# 生成问题并打印
question, answer = generate_question()
print(question)
print(answer)