可以使用Python的os模块来比较文件名并将其添加到列表中。下面是一个示例代码:
import os
# 指定目录路径
directory = "/path/to/directory"
# 初始化一个空列表
file_list = []
# 遍历目录中的所有文件
for filename in os.listdir(directory):
# 如果文件名以特定的后缀结尾(例如".txt"),可以使用字符串的endswith()方法进行比较
if filename.endswith(".txt"):
# 将符合条件的文件名添加到列表中
file_list.append(filename)
# 打印列表中的文件名
print(file_list)
上述代码中,首先需要指定要遍历的目录路径,将其赋值给变量directory
。然后创建一个空列表file_list
用于存储符合条件的文件名。
接下来使用os.listdir()
函数遍历目录中的所有文件,通过endswith()
方法比较文件名是否以指定的后缀结尾。如果符合条件,则将文件名添加到列表中。
最后,使用print()
函数打印列表中的文件名。
请注意,上述代码只是一个示例,你可以根据自己的需求进行修改和扩展。