Python实用小呆萌 之 获取当前目录下文件夹及子文件夹下的文件名

Python实用小呆萌 之 获取当前目录下文件夹及子文件夹下的文件名

1. 获取目录下文件名

import os
path = "D:\\LEON\\Sevenstar\\Files\\"  # path为文件夹路径
file_list = os.listdir(path)  # file_list即为路径下文件夹名或文件名
print(file_list)

2. 获取目录下文件夹下 子文件夹内的文件名

获取当前文件夹下，子文件夹内的文件名（着实拗口）思路就是先获取当前文件夹下的文件夹名，再循环打开文件夹获取子文件夹里面的文件名（请忽略这些话，直接看代码）

# 获取文件夹下所有文件，包含子文件夹
import os
path = "D:\\LEON\\Sevenstar\\Files\\"
file_list_2D = []
file_list_pre = os.listdir(path)  # 获取当前文件夹下的文件夹名
print(file_list_pre)


try:
    for i in file_list_pre:  # 依次打开子文件夹，获取子文件夹的文件名。这将是一个二维列表
        all_list = os.listdir(path + i)
        file_list_2D.append(all_list)

    print(file_list_2D)
except:
    pass
#  使用try的原因是避免当前目录下有单独文件，而非文件夹时产生的错误


#  如果只需要所有文件的列表，可以将二维列表转化为一维列表
all_file = sum(file_list_2D, [])
print(all_file)

感觉这种方法是一种笨拙的方法，而且如果子文件夹下还有文件夹就不能获取到这个文件夹里面的文件了，相信还会有更简便快捷的方法的。