扫描二维码下载沐宇APP

沐宇

微信扫码使用沐宇小程序

沐宇

python怎么读取目录下所有文件名

扬州沐宇科技
2023-12-01 19:48:28
python

可以使用os模块来读取目录下的所有文件名。具体方法如下:

import os

def get_file_names(directory):
    file_names = []
    for filename in os.listdir(directory):
        if os.path.isfile(os.path.join(directory, filename)):
            file_names.append(filename)
    return file_names

# 读取当前目录下的所有文件名
current_directory = os.getcwd()
files = get_file_names(current_directory)
print(files)

上述代码中,get_file_names函数接受一个目录路径作为参数,遍历目录下的所有文件和文件夹,通过os.path.isfile函数判断是否为文件,将文件名添加到file_names列表中。最后输出所有文件名。

如果需要递归地读取目录下的所有文件名,可以使用os.walk函数,示例如下:

import os

def get_file_names(directory):
    file_names = []
    for root, dirs, files in os.walk(directory):
        for filename in files:
            file_names.append(filename)
    return file_names

# 递归地读取当前目录下的所有文件名
current_directory = os.getcwd()
files = get_file_names(current_directory)
print(files)

扫码添加客服微信