Python random实战：双色球抽奖

import random
number = input("请输入数量:").strip()


if not number.isdigit():
    print("只能输入数字")
else:
    number = int(number)
    all_data = set() #这个集合是存放所有的号码
    red_ball_range = [str(i).zfill(2) for i in range(1, 34)] #红色球范围
    print(red_ball_range)
    blue_ball_range = [str(i).zfill(2) for i in range(1, 17)]#蓝色球范围
    while len(all_data) != number:
        red_ball = random.sample(red_ball_range,6)
        red_ball.sort()
        print(red_ball)
        blue_ball = random.choice(blue_ball_range)
        red_ball.append(blue_ball)
        ssq = " ".join(red_ball) + '\n' #最后生成的结果
        print(ssq)
        all_data.add(ssq)
    with open("ssq.txt","w") as fw:
        fw.writelines(all_data)