用Python实现筛选文件脚本的方法_Python

用Python实现筛选文件脚本的方法

2021-04-13 00:42林下的码路 Python

今天小编就为大家分享一篇用Python实现筛选文件脚本的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

在做项目时遇到需要标记数据集里面的若干图片数据，作为程序员，为避免手动一张一张的筛选，所以写了这个Python 脚本实现。

Python脚本如下：

									# from PIL import Image

									import csv

									import os

									import shutil

									filename = 'img.txt'

									def readImageName():

									 with open(filename) as f:

									  lines = f.readlines()

									  imgnames = []

									  for line in lines:

									   imgnames.append(line.strip().strip(".jpg")[-4:])

									  print(imgnames)

									  return imgnames

									def pickImg():

									 pickImageNames = readImageName()

									 # 遍历所有图片集的文件名

									 for image in os.listdir(r"C:\Users\Administrator.PC-201708272051\Desktop\项目组\text_detect_label_data\China_SameBrowser"):

									  # print(image[:-4])

									  if image[:-4] in pickImageNames:

									   # pickImage = Image.open((r"C:\Users\Administrator.PC-201708272051\Desktop\项目组\text_detect_label_data\China_SameBrowser/%s") % image)

									   # pickImage.save((r"C:/Users/Administrator.PC-201708272051/Desktop/labeldata/%s") % image)

									   oldname= r"C:\Users\Administrator.PC-201708272051\Desktop\项目组\text_detect_label_data\China_SameBrowser/" + image

									   newname= r"C:/Users/Administrator.PC-201708272051/Desktop/labeldata/" + image

									   shutil.copyfile(oldname,newname)

									# readImageName()

									pickImg()