王剑编程网

分享专业编程知识与实战技巧

编写一个自动生成双色球号码的 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())

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

控制面板
您好,欢迎到访网站!
  查看权限
网站分类
最新留言