Python妙用|给小外甥生成10以内加减运算数学做作业

import random
from datetime import datetime

#加、减模板
add_formula = '{a} + {b} = {c}'
sub_formula = '{a} - {b} = {c}'

formulas = set()

for i in range(50):
    a=random.randint(0, 10)
    b=random.randint(0, 10)
    formulas.add(add_formula.format(a=a, b=b, c=a+b))
    if a>=b:
        #小外甥这个岁数没有负数的概念，需要满足a>=b的减法
        formulas.add(sub_formula.format(a=a, b=b, c=a-b))

#习题去重且有序，方便做题后自己检查对错
formulas = list(formulas)

#保存到加减md中，方便导出pdf
month = datetime.today().month
day = datetime.today().day
with open('加减.md', 'w', encoding='utf-8') as f:
    f.write('## {month}月{day}号 加减运算\n'.format(month=month, day=day))
    for fm in formulas:
        f.write(fm.split('= ')[0] + ' =\n\n\n')

        
with open('加减答案.md', 'w', encoding='utf-8') as f:
    f.write('## {month}月{day}号 加减运算\n'.format(month=month, day=day))
    for fm in formulas:
        f.write(fm + '\n\n')