如何修正Python程序嵌套循环以获取指定数量的质数

问题根源 #

你的核心问题是逻辑方向错误：原is_prime函数是找出2到user_number之间的所有质数，而非获取前user_number个质数。输入100时，它只会返回2到99之间的25个质数，自然无法得到第100个质数（541）。

另外format_primes函数存在无效赋值：printed_primes = print(...)毫无意义，因为print()函数返回None。

修正后的代码 #

def get_n_primes(n):
    # 获取前n个质数
    primes = []
    num = 2
    while len(primes) < n:
        # 判断当前数是否为质数
        is_prime = True
        # 优化：只需检查到当前数的平方根即可
        for x in range(2, int(num**0.5) + 1):
            if num % x == 0:
                is_prime = False
                break
        if is_prime:
            primes.append(num)
        num += 1
    return primes

def format_primes(primes_list):
    # 按要求格式打印质数
    for sequence, prime in enumerate(primes_list, start=1):
        print(f"Prime # {sequence} = {prime}")

def get_user_num():
    # 补充用户输入获取逻辑（原代码缺失）
    while True:
        try:
            num = int(input("请输入要获取的质数数量："))
            if num > 0:
                return num
            print("请输入正整数！")
        except ValueError:
            print("输入无效，请输入整数！")

def main():
    user_number = get_user_num()
    list_of_primes = get_n_primes(user_number)
    format_primes(list_of_primes)

if __name__ == "__main__":
    main()

关键改动说明 #

• 重构质数获取逻辑：将原is_prime改为get_n_primes，通过while循环持续寻找质数，直到列表长度达到用户指定的数量
• 质数判断优化：仅检查到当前数的平方根，减少不必要的循环计算
• 修复打印函数：去掉无效变量赋值，直接循环输出符合格式的内容
• 补充缺失的get_user_num函数，确保程序能正常接收用户输入

测试效果 #

输入100后，程序会输出前100个质数，最后一行即为Prime # 100 = 541，完全符合需求。

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