Nate_2020 发表于 2021-2-23 13:46:12

递归函数返回问题

借一鱼油的问题, 想得到结果, 于是需要这个递归函数返回结果, 但不通知怎么弄, 请大神帮忙
我现在是想让这个函数返回这么目标文件的路径,
因为可能有多个, 可以用列表的形式返回,
这能实现返回吗?

递归函数如下


import os

def search_file(start_dir, target) :
    os.chdir(start_dir)
   
    for each_file in os.listdir(os.curdir) :
      if each_file == target :
            print(os.getcwd() + os.sep + each_file) # 使用os.sep是程序更标准
      if os.path.isdir(each_file) :
            search_file(each_file, target) # 递归调用
            os.chdir(os.pardir) # 递归调用后切记返回上一层目录

start_dir = input('请输入待查找的初始目录:')
target = input('请输入需要查找的目标文件:')
search_file(start_dir, target)

笨鸟学飞 发表于 2021-2-23 13:57:55

你建个全局变量,把结果加进去不就完了么

Stubborn 发表于 2021-2-23 14:04:39

意思是返回目标文件的完整路径,目标文件可能存在多个?

def search_file(start_dir, target, result=None):
    if result is None:
      result = []
    os.chdir(start_dir)
    for each_file in os.listdir(os.curdir):
      if each_file == target:
            result.append(os.path.join(os.getcwd(), each_file))
      if os.path.isdir(each_file):
            search_file(each_file, target, result)
            os.chdir(os.pardir)
    return result

jackz007 发表于 2021-2-23 14:17:15

本帖最后由 jackz007 于 2021-2-23 16:11 编辑

import os

def search_file(start_dir , target) :
    result = []
    try:
      for each_file in os . listdir(start_dir) :
            x = os . path . join(start_dir , each_file)
            if os . path . isfile(x) :
                if target . lower() == each_file . lower() :
                  result . append(x)
            elif os . path . isdir(x) :
                result += search_file(x , target)
    except Exception as e:
      print(e)
    return result

start_dir = input('请输入待查找的初始目录:') . strip()
if start_dir :
    target = input('请输入需要查找的目标文件:') . strip()
    if target :
      r = search_file(start_dir , target)
      if r :
            for x in r:
                print(x)
      else :
            print('抱歉,没有找到目标文件。')

Nate_2020 发表于 2021-2-24 16:52:26

笨鸟学飞 发表于 2021-2-23 13:57
你建个全局变量,把结果加进去不就完了么

it's nice. thanks

Nate_2020 发表于 2021-2-24 16:53:01

jackz007 发表于 2021-2-23 14:17


good
页: [1]
查看完整版本: 递归函数返回问题