如何使用Python的gmpy2库生成指定N位长度的随机整数

原代码问题分析 #

你当前的代码逻辑错误，*gmpy2.mpz_random()*的第二个参数是随机数的上界（不包含），传入n=100只会生成0~99之间的整数，完全无法实现生成100位大整数的需求。

实现逻辑 #

标准的N位十进制整数的取值范围为[10^(N-1), 10^N - 1]，天然没有前导零，刚好匹配你的需求。我们可以先计算出该区间的长度为9*10^(N-1)，调用mpz_random生成该长度范围内的随机数后，加上区间最小值10^(N-1)，即可得到符合要求的N位随机整数。

修改后可直接运行的代码 #

import gmpy2

def generate_n_digit_random(n):
    if n < 1:
        raise ValueError("输入的位数n必须大于等于1")
    # 初始化随机种子
    rs = gmpy2.random_state(hash(gmpy2.random_state()))
    # 计算区间最小值和区间长度
    min_val = gmpy2.mpz(10) ** (n - 1)
    range_length = (gmpy2.mpz(10) ** n) - min_val
    # 生成并返回目标随机数
    return gmpy2.mpz_random(rs, range_length) + min_val

# 测试用例
print(generate_n_digit_random(1))
print(generate_n_digit_random(2))
print(generate_n_digit_random(4))
print(generate_n_digit_random(100))

效果说明 #

运行代码后输出完全匹配你给出的示例要求：

• 输入n=1时输出1~9的1位整数
• 输入n=2时输出10~99的2位整数
• 输入n=100时输出刚好100位的大整数，不会出现前导零或者位数不足的情况。

内容的提问来源于stack exchange，提问作者Ahmed Khan