[已解决]python 递归调用函数

二爷zou

1. def search_file(start_dir):
   
2.     os.chdir(start_dir)
   
3.     print(os.getcwd())
   
4.     target = ['.c', '.cpp', '.py', '.cc', '.java', '.pas', '.asm']
   
5.     line_num = []
   
6.     sum = 0
   
7. 
   
8.     for each_file in os.listdir():
   
9.         ext = os.path.splitext(each_file)[1]
   
10.         if ext in target:
    
11.             line_num.append(calc_code(each_file))
    
12.             print(line_num)
    
13.                
    
14.         if os.path.isdir(each_file):
    
15.             search_file(os.getcwd() + '/' +each_file)
    
16.             os.chdir('..')
    
17. 
    
18.     for j in range(len(line_num)):
    
19.         sum += int(line_num[j])
    
20.     return sum

复制代码

递归调用函数，找出指定路径下所有文件的总行数， 遇到文件夹也需要计算。

也就是说如果我想把递归调用函数的出的结果统一保存， 可是每递归调用一次函数， 起那面的line_num 列表就是空的， sum 也是等于0 的。这个有什么办法解决没有。

最佳答案

月排行榜 / 总排行榜

qiuyouzhi

2020-3-13 16:39:13

把line_num和sum放到函数的外面

跳转到最佳答案楼层

qiuyouzhi

把line_num和sum放到函数的外面

二爷zou

qiuyouzhi 发表于 2020-3-13 16:39
把line_num和sum放到函数的外面

放到函数外边还能生效吗

sunrise085

在第二个if中加了一行代码，将递归的返回值也添加到列表中，即将下一层文件夹中的文件总行数返回一并统计到该层文件夹的文件总行数中。我没有测试，你试一下是否正确。

1. def search_file(start_dir):
   
2.     os.chdir(start_dir)
   
3.     print(os.getcwd())
   
4.     target = ['.c', '.cpp', '.py', '.cc', '.java', '.pas', '.asm']
   
5.     line_num = []
   
6.     sum = 0
   
7. 
   
8.     for each_file in os.listdir():
   
9.         ext = os.path.splitext(each_file)[1]
   
10.         if ext in target:
    
11.             line_num.append(calc_code(each_file))
    
12.             print(line_num)
    
13.                
    
14.         if os.path.isdir(each_file):
    
15.             lines=search_file(os.getcwd() + '/' +each_file)
    
16.             line_num.append(lines)
    
17.             os.chdir('..')
    
18. 
    
19.     for j in range(len(line_num)):
    
20.         sum += int(line_num[j])
    
21.     return sum

复制代码

qiuyouzhi

二爷zou 发表于 2020-3-13 16:44
放到函数外边还能生效吗

能的

二爷zou

qiuyouzhi 发表于 2020-3-13 16:50
能的

import easygui as g
import os

def calc_code(file_name):
    lines = 0
    with open(file_name) as f:
        print("正在分析文件： %s..." % file_name)
        try:
            for each_line in f:
                lines += 1
        except UnicodeDecodeError:
            pass
    return lines


def search_file(start_dir):
    os.chdir(start_dir)
    print(os.getcwd())
    target = ['.c', '.cpp', '.py', '.cc', '.java', '.pas', '.asm']
   

    for each_file in os.listdir():
        ext = os.path.splitext(each_file)[1]
        if ext in target:
            line_num.append(calc_code(each_file))
               
        if os.path.isdir(each_file):
            search_file(os.getcwd() + os.sep + each_file)
            os.chdir(os.pardir)




line_num = []
sum = 0
start_dir = g.diropenbox()
search_file(start_dir)
for j in range(len(line_num)):
    sum += int(line_num[j])
print(sum)




是可以的！ 感谢！  最终这样写能得到指定目录下的总行数

qiuyouzhi

二爷zou 发表于 2020-3-13 17:02
import easygui as g
import os

如果有帮助，请设最佳答案

二爷zou

sunrise085 发表于 2020-3-13 16:45
在第二个if中加了一行代码，将递归的返回值也添加到列表中，即将下一层文件夹中的文件总行数返回一并统计到 ...

这个写法好像不太行呦， 空列表和num 值，只要递归调用这个函数一遍就会变成空值和0   

二爷zou

qiuyouzhi 发表于 2020-3-13 17:13
如果有帮助，请设最佳答案

已经设了  哈哈哈
感谢
再次感谢