Python 生成 4 位随机数的多种实现方法与最佳实践

实现步骤

1. 导入 random 和 string 模块。
2. 获取所有数字字符（0-9）。
3. 循环选择 4 次并拼接。

import random
import string

# 获取所有数字字符 '0123456789'
digits = string.digits

# 使用列表推导式或 join 生成 4 位随机字符串
random_code = ''.join(random.choice(digits) for _ in range(4))
print(random_code)

优点： 此方法生成的可能是以 0 开头的数字（如 "0123"），这在某些验证码场景中是允许的，而 randint 无法生成以 0 开头的四位数。

4. 安全场景下的最佳实践：secrets 模块

重要提示： 上述 random 模块生成的随机数是伪随机数，基于算法种子生成，不适合用于安全敏感的场景（如密码重置令牌、API Key、会话 ID 等）。攻击者可能通过预测种子来重现序列。

Python 3.6+ 引入了 secrets 模块，它基于操作系统的加密源（如 /dev/urandom 或 CryptGenRandom），提供加密安全的随机数。

使用 secrets 生成 4 位随机数

import secrets

# 生成 1000 到 9999 之间的安全随机整数
secure_number = secrets.randbelow(9000) + 1000
print(secure_number)

或者生成 4 位安全字符串：

import secrets
import string

digits = string.digits
secure_code = ''.join(secrets.choice(digits) for _ in range(4))
print(secure_code)

建议： 如果是用于验证码、登录 Token 或密码，请务必使用 secrets 模块替代 random。

5. 控制随机种子 (Seed)

在某些调试或复现场景下，我们需要让随机数生成过程是可重复的。这可以通过设置种子（seed）来实现。

import random

random.seed(42)  # 设置固定种子
print(random.randint(1000, 9999))  # 每次运行输出相同

random.seed(42)
print(random.randint(1000, 9999))  # 再次输出相同

在生产环境中，通常不建议固定种子，除非是为了单元测试的可复现性。

6. 总结

在实际开发中，请根据具体需求选择合适的方案。对于普通的业务逻辑，random 模块足够高效；涉及用户安全信息时，必须切换到 secrets 模块以确保系统安全。