可以使用os.listdir()
函数来获取指定目录下的所有文件名。这个函数会返回一个包含目录中所有文件和目录名称的列表。例如,以下代码将获取当前目录(“.”)下的所有文件名:
import os
file_names = os.listdir(".")
print(file_names)
如果要获取其他目录下的文件名,只需将目录路径作为参数传递给os.listdir()
函数即可。例如,以下代码将获取/path/to/directory
目录下的所有文件名:
import os
directory = "/path/to/directory"
file_names = os.listdir(directory)
print(file_names)
注意:os.listdir()
函数将返回目录中所有文件和目录的名称,包括隐藏文件和目录。如果只需要获取文件名而不包括子目录和隐藏文件,可以使用os.path.isfile()
函数来过滤结果。以下代码将只获取指定目录下的文件名:
import os
directory = "/path/to/directory"
file_names = [file for file in os.listdir(directory) if os.path.isfile(file)]
print(file_names)