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