利用python 获取当前文件夹下所有文件名

这里通过os下的两个函数实现,os.walk()和os.listdir()。

import os  
  
def file_name(file_dir):   
    for root, dirs, files in os.walk(file_dir):  
        print(root) #当前目录路径  
        print(dirs) #当前路径下所有子目录  
        print(files) #当前路径下所有非目录子文件

可以查找指定类型的文件

import os  
  
def file_name(file_dir):   
    L=[]   
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] == '.csv':  
                L.append(os.path.join(root, file))  
    return L  

#其中os.path.splitext()函数将路径拆分为文件名+扩展名

 

你可能感兴趣的:(自学)