数据与总结
文件夹下有1000张图,但是需要筛选符合条件的图片到另一个文件夹。
数据:
1、装有1000张图的文件夹
2、一个csv文本文件,需要取出里面的plot_no这一列,匹配符合的图片
解决:
1、图片批量复制的问题
2、python中if,条件比较需要注意数据类型。不确定的可以type()查看类型(这个问题经常犯错故总结)
tip:我的图片命名为1.jpg~1000.jpg,csv文本文件中plot_no列中的数据为6,8,9,22这样,故实现的是保存满足plot_no的图片,复制。
代码实现
import pandas as pd
import shutil
import os
path = r"E:\zzz\zz2"
pathDir = os.listdir(path)
path2_2 = r'E:\zzz\zzzz\西部.csv'
data = pd.read_csv(path2_2)
for i in pathDir:
for j in data.plot_no:
file_name_1 = os.path.basename(i)
file_name2_2 = file_name_1.split('.')[0] ###获取照片的名
#print(type(file_name2_2),type(j))
if str(file_name2_2)==str(j):###注意判断的时候数据类型需要一致
print(file_name2_2)
shutil.copy(r"E:\***\***\%s.jpg"% file_name2_2, r'E:\***\***\%s.jpg' % file_name2_2)
###后缀为.jpg图片批量复制,.csv文本文件批量复制等