我有一个包含数千张图片的文件夹。有些图像有一个同名的txt文件。例如abc.jpg、abc.txt。我想保留这些图像,并删除所有没有txt文件的图像。有办法做到这一点吗?
这将把所有.jpg
,.png
,.jpeg
和它们各自的.txt
文件移动到一个新文件夹。
import os
import shutil
folderdir = "folder_path"
dirs = os.listdir(folderdir)
for i in dirs:
if i.endswith(".txt"):
txt = i
jpg = str(i.split(".txt")[0]) + ".jpg"
png = str(i.split(".txt")[0]) + ".png"
jpeg = str(i.split(".txt")[0]) + ".jpeg"
try:
shutil.move(folderdir + "/" + jpg, 'dataset')
shutil.move(folderdir + "/" + txt, 'dataset')
except:
try:
shutil.move(folderdir + "/" + png, 'dataset')
shutil.move(folderdir + "/" + txt, 'dataset')
except:
shutil.move(folderdir + "/" + jpeg, 'dataset')
shutil.move(folderdir + "/" + txt, 'dataset')
print("DONE")