温馨提示×

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

小亿
105
2023-12-01 19:48:28
栏目: 编程语言

可以使用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)

0