Python函数返回None引发'NoneType不可迭代'错误求助

首先，咱们先揪出你遇到的NoneType object is not iterable错误的根源：你的prime_func()只有在counter == args[1]的时候才会返回值，一旦循环结束都没触发这个条件，函数就会默认返回None——而你试图把None解包成三个变量，这肯定会报错。

另外，你的质数判断逻辑也有问题，比如x=4的时候，y=2会把它加入合数列表，但y=3的时候又会把它加入质数列表，这明显不符合质数的定义。还有循环只到50，要是你要找的质数超过50以内的数量，直接就会返回None。

下面是修复后的完整代码，我会一步步解释改动点：

def find_nth_prime(n):
    # 初始化质数列表，前两个质数是2和3
    primes = [2, 3]
    composites = []
    
    # 处理n为1或2的边界情况
    if n == 1:
        return 1, composites, [2]
    if n == 2:
        return 2, composites, primes
    
    counter = 2  # 已经统计了2个质数
    current_num = 4  # 从4开始检查每个数
    
    # 循环直到找到第n个质数
    while counter < n:
        is_prime = True
        # 优化：只需检查到当前数的平方根，不用遍历到current_num-1
        for divisor in range(2, int(current_num**0.5) + 1):
            if current_num % divisor == 0:
                composites.append(current_num)
                is_prime = False
                break  # 找到因数就停止检查，节省资源
        
        if is_prime:
            primes.append(current_num)
            counter += 1
        
        current_num += 1
    
    # 无论如何都会返回三个值，不会出现None
    return counter, composites, primes

# 调用示例：找第7个质数
target_prime = 7
counter1, composites1, primes1 = find_nth_prime(target_prime)
print(f"第{target_prime}个质数是: {primes1[-1]}")
print(f"总共找到{counter1}个质数")
print(f"排查过的合数列表: {composites1}")

关键修复点说明： #

• 解决返回值问题：
  现在函数在任何场景下都会返回三个值——不管是边界情况（n=1/2）还是循环找到目标质数后，再也不会出现返回None的情况，彻底解决了解包错误。

• 修正质数判断逻辑：
  用current_num % divisor == 0判断整除，比你原来的除法后比较整数更准确，避免浮点数精度问题；一旦找到能整除的因数，立刻标记为合数并跳出循环，不用做无用的遍历；同时只检查到当前数的平方根，这是质数判断的经典优化，能大幅提升效率。

• 变量作用域优化：
  你之前纠结变量是否该放在函数内——其实局部变量才是更优的选择，把primes、composites、counter都放在函数内部，避免全局变量被意外修改，也让函数的封装性更好。

• 灵活的循环逻辑：
  原来的循环只到50，现在改成while counter < n，会一直检查直到找到第n个质数，不管这个质数多大。

• 明确的输入参数：
  把原来依赖的args[1]改成函数参数n，让函数的输入更清晰，调用起来也更方便，不用依赖外部的全局变量。

最后，你可以根据自己的需求调整调用时的target_prime值，比如改成10就能拿到第10个质数29。

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