编写一个自动生成双色球号码的 Python 小脚本

双色球号码很难选，如果你没有坚守的号码，一般都是机选。下面用 Python 编程语言写一个机选号码小脚本，给生活增加一些乐趣。

首先，双色球的规则如下：

• 双色球分为红球和蓝球
• 红球有 6 个号码，从 1-33 中随机选择，顺序不重要
• 蓝球有 1 个号码，从 1-16 中随机选择

要产生随机数，需要使用 Python 的 random 模块。其中的 random.randint(a, b) 可以生成 [a, b] 闭区间的随机整数。因此，生成蓝球的代码如下：

# 导入 random 模块
import random

# 产生一个 1-16 之间的随机整数，包含首尾的 1 和 16
blue = random.randint(1, 16)

random.sample(list, n) 方法从 list 列表中随机抓取 n 个元素。比如，下面代码，从数字 1-6 中随机选 3 个数字：

random.sample([1, 2, 3, 4, 5, 6], 3)

Python 为了简化数字序列的创建，提供了 range(start, end) 函数。它会创建一个从 start 开始，到 end 结束（不包含 end），依次递增的序列。比如 range(1, 7) 表示的序列和 [1, 2, 3, 4, 5, 6] 范围相同。

因此，红球的生成逻辑如下：

# 生成一个 1-33 的数字序列
nums = range(1, 34)
# 从上述序列中随机挑选 6 个元素
reds = random.sample(nums, 6)

上述代码可以合并为一条代码：

reds = random.sample(range(1, 34), 6)

此时的 reds 包含随机挑选的数字序列，比如 [17, 23, 11, 9, 30, 31]。因为红球的顺序不重要，为了使其美观，可以使用 sorted(list) 函数对数字序列排序。

reds = sorted(reds)
# 排序后，数字从小到大排列，比如：
# [9, 11, 17, 23, 30, 31]

至此，一组双色球号码就创建成功了。把它封装为函数，方便复用。

Python 使用 def 关键字定义函数，函数体的代码块使用缩进表示。官方建议一级缩进包含 4 个空格。

Python 函数可以返回逗号分隔的多个值，这些值会被 Python 放在一个元组类型（tuple）的变量中。元组和 JavaScript 的数组类似，只不过元组是只读的，内容不能被修改。

import random

# 创建一组双色球号码
def union_lottery():
    blue = random.randint(1, 16)
    reds = random.sample(range(1, 34), 6)
    reds = sorted(reds)
    return reds, blue

# 打印双色球号码    
print(union_lottery())

把代码保存到 app.py 中，在终端执行 python app.py，可以看到类似如下的输出（因为都是随机数，每次输出都不一样）：

([4, 8, 19, 20, 22, 33], 16)

方括号中的是红球数字，后面的是蓝球数字。

现在每次执行只能输出一组号码，如果希望用户可以指定数量，可以使用 input(msg) 函数获取用户输入的值，其中的 msg 是提示语。比如：

name = input("你叫什么名字？")
print(f"你好，{name}！")

字符串前的 f 表示它不是普通字符串，它可以使用变量插值，就像 JavaScript 的模板字符串那样。

input() 的返回值是字符串，使用 int() 函数把它转换为整数类型。拿到用户输入的数字后，就可以通过循环打印指定的双色球。

num = input('你想生成几组号码？')
for i in range(int(num)):
    print(union_lottery())

现在功能实现了，但是太丑了，需要美化外观。

我们想用颜色区分红球和蓝球。此处只需要两种颜色，没必要引入第三方库，直接使用 ANSI escape codes 定义两个颜色函数：

def red(msg):
    return f'\x1b[1;31m{msg}\x1b[0m'

def blue(msg):
    return f'\x1b[1;34m{msg}\x1b[0m'

对函数返回值（元组类型）解包，得到红球和蓝球：

red_balls, blue_ball = union_lottery()

红球是一个列表，使用 map(func, list) 为每个数字添加红色。蓝球是个单一数字，直接调用函数就能添加蓝色。

red_balls = map(red, red_balls)
blue_ball = blue(blue_ball)

所有数字已经带颜色，使用制表符 \t 把它们连接。字符串的 join(list) 方法用于连接列表。

print('\t'.join(red_blues), '\t', blue_ball)

效果如下：

可以把美化相关代码提取到一个函数 pretty_print 中：

def pretty_print(nums):
    red_balls, blue_ball = nums
    red_balls = map(red, red_balls)
    blue_ball = blue(blue_ball)
    print('\t'.join(red_balls), '\t', blue_ball)

num = input('你想生成几组号码？')
for i in range(int(num)):
    pretty_print(union_lottery())

双色球生成器的整体代码如下：

import random

def union_lottery():
    blue = random.randint(1, 16)
    reds = random.sample(range(1, 34), 6)
    reds = sorted(reds)
    return reds, blue

def pretty_print(nums):
    red_balls, blue_ball = nums
    red_balls = map(red, red_balls)
    blue_ball = blue(blue_ball)
    print('\t'.join(red_balls), '\t', blue_ball)

def red(msg):
    return f'\x1b[1;31m{msg}\x1b[0m'

def blue(msg):
    return f'\x1b[1;34m{msg}\x1b[0m'

num = input('你想生成几组号码？')
for i in range(int(num)):
    pretty_print(union_lottery())

最后，彩票有风险，投注需谨慎。