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