我具有以下目录结构和以下文件:
Folder_One
├─file1.txt
├─file1.doc
└─file2.txt
Folder_Two
├─file2.txt
├─file2.doc
└─file3.txt
我只想从列出的每个文件夹中获取.txt文件。例:
Folder_One-> file1.txt and file2.txt
Folder_Two-> file2.txt and file3.txt
注意:整个目录位于一个名为“数据集”的文件夹中。我的代码看起来像这样,但是我相信有些东西丢失了。有人能帮我吗。
path_dataset = "./dataset/"
filedataset = os.listdir(path_dataset)
for i in filedataset:
pasta = ''
pasta = pasta.join(i)
for file in glob.glob(path_dataset+"*.txt"):
print(file)
您可以使用re
模块检查文件名是否以结尾.txt
。
import re
import os
path_dataset = "./dataset/"
l = os.listdir(path_dataset)
for e in l:
if os.path.isdir("./dataset/" + e):
ll = os.listdir(path_dataset + e)
for file in ll:
if re.match(r".*\.txt$", file):
print(e + '->' + file)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句