其实函数的递归调用不是特别的常用,因为递归的速度会稍微慢一点,而且python本身对于递归的次数(或者说层数)是有要求的,大概小于1000。后面我会提到这个限制以及如何设置上限的问题。不过就算可以设置上限,在实际应用中也会比较麻烦。
不过,函数的递归调用对于解决某些问题是非常非常方便的。
我们以最经典的递归例子:求阶乘为例。
def jiecheng(n):
n = int(n)
if n == 0: #如果不设置这个条件的话,可能会陷入死循环
return 1
else:
return n * jiecheng(int(n-1))
print(jiecheng(3))
### 6
大家可以尝试一下当你把n设置成1000的时候,程序就会报错。为了解决这个问题,我们可以用以下代码。这里的10000是可以自己设置的。这样就设置了上限。
import sys
sys.setrecursionlimit(10000)
最后来看一下如何遍历一个文件夹下所有的文件。思路是首先获取文件夹下第一层级文件和文件夹的名称,然后判断如果是文件则输出,如果是文件夹则递归调用自己。
import os
def get_file(path):
filenames = os.listdir(path)
for file in filenames:
child = os.path.join(path,file)
if os.path.isdir(child):
get_file(child)
else:
print(child)
input_dir = r'C:\Users\Mr.Reliable\Desktop\PyTorch-Tutorial'
print(get_file(input_dir))