Python递归调用返回None的常见陷阱与修复

递归调用中“打印有值，返回None”的原因

在编写递归函数时，开发者常遇到一种情况：函数内部正确打印了期望的结果，但函数调用后返回的值却是 None。这通常是因为在递归调用的地方遗漏了 return 语句，导致数据在递归层之间无法正确传递回最外层调用。

实例一：字符串右移操作

假设我们需要实现一个字符串右移函数 right_shift(s, n)，将字符串前 n 个字符移动到末尾。以下是一个存在问题的实现：

s1 = 'abcdefg'

def right_shift(s, n):
    """
    把传入的字符串,前n个字符移动到最后面
    """
    if n < 1:
        print(s) # 此步输出结果为 "efgabcd"
        return s
    s = s[1:] + s[0]
    n -= 1
    right_shift(s, n)

s = right_shift(s1, 4)
print(s)# 此步输出结果为 None

在上述代码中，当 n < 1 时，函数正确返回了处理后的字符串。然而，在递归调用 right_shift(s, n) 时，没有使用 return。这意味着递归调用返回的值被丢弃了，最外层的函数调用最终返回 None。

修复方案

解决此问题的方法是在递归调用前加上 return 语句，确保递归层的返回值能逐层传递回最外层：

s1 = 'abcdefg'
def right_shift(s, n):
    """
    把传入的字符串,前n个字符移动到最后面
    """
    if n < 1:
        # print(s)
        return s
    s = s[1:] + s[0]
    n -= 1
    # 加入return, 其原理根装饰器的闭包接近!
    return right_shift(s, n)
s = right_shift(s1, 4)
print(s) # 成功输出 "efgabcd"

实例二：对象层级遍历

在另一个实际场景中，我们需要获取某个对象所在的原始父类。以下代码同样存在递归返回值丢失的问题：

 def get_end_parent_ele(self, obj):
    """获取这个id 所在的原始类"""
    parent_id = obj.parent_id if obj else 0
    OHHOLog.print_log("获取这个id parent_id")
    OHHOLog.print_log(parent_id)
    parent_obj = self.obj_obj.get_by_id(parent_id)
    if parent_obj.key == "all":
      OHHOLog.print_log("all")
      return_key = obj.key
      OHHOLog.print_log(return_key)
      return return_key
    else:
      self.get_end_parent_ele(parent_obj)

当 parent_obj.key != "all" 时，函数递归调用 self.get_end_parent_ele(parent_obj)，但未返回该调用的结果。这会导致最终返回值为 None。

修复方案

在递归调用的地方添加 return 语句，确保正确返回最终结果：

 def get_end_parent_ele(self, obj):
    """获取这个id 所在的原始类"""
    parent_id = obj.parent_id if obj else 0
    OHHOLog.print_log("获取这个id parent_id")
    OHHOLog.print_log(parent_id)
    parent_obj = self.interest.get_by_id(parent_id)
    if parent_obj.key == "all":
      OHHOLog.print_log("all")
      return_key = obj.key
      OHHOLog.print_log(return_key)
      return return_key
    else:
      return self.get_end_parent_ele(parent_obj)

总结

在Python递归函数中，务必确保所有递归调用都使用 return 语句。否则，即使内部逻辑正确，最终返回值也会是 None。通过仔细检查递归调用的返回值传递，可以有效避免此类隐蔽错误。

以上就是Python递归调用返回None的常见陷阱与修复的详细内容，更多关于Python递归调用的资料请关注本站其它相关文章！