python修改csv文件某一列统一加后缀

import numpy as np
import pandas as pd
test_pic=pd.read_csv("D:\\Users\\sgcai\\Documents\\term_01_nima\\csg\\lvpai\\test.csv")
test_pic['pic id']=test_pic['pic id'].apply(lambda x:str(int(x))+'.jpg')
test_pic.to_csv("D:\\Users\\sgcai\\Documents\\term_01_nima\\csg\\lvpai\\test_v1.csv",index=None)

提取图片名称,对两个文件夹的图片进行搜索并查找制定特征的图片,最后保存到另一个文件夹当中

import os
from PIL import Image
dir_img="D:\\Users\\sgcai\\Documents\\term_01_nima\\csg\\lvpai\\buhaokan5w\\"
dir_img_haokan="D:\\Users\\sgcai\\Documents\\term_01_nima\\csg\\lvpai\\haokan\\"
dir_save="D:\\Users\\sgcai\\Documents\\term_01_nima\\csg\\lvpai\\test_file\\"
size=(256,192)

list_img_buhaokan = os.listdir(dir_img)#获取目录下所有图片名
list_img_haokan = os.listdir(dir_img_haokan)#获取目录下所有图片名
test_listname=pd.read_csv("D:\\Users\\sgcai\\Documents\\term_01_nima\\csg\\lvpai\\test_v1.csv")
#遍历
cnt=0
print(len(list_img))
print(len(test_listname['pic id']))
for pic_id in test_listname['pic id']:
    if pic_id in list_img_buhaokan:
        pri_image = Image.open(dir_img+pic_id)
        tmppath=dir_save+pic_id
        pri_image.save(tmppath)
    elif pic_id in list_img_haokan:
        pri_image_haokan = Image.open(dir_img_haokan+pic_id)
        tmppath=dir_save+pic_id
        pri_image_haokan.save(tmppath)
发布了373 篇原创文章 · 获赞 18 · 访问量 9万+

猜你喜欢

转载自blog.csdn.net/csg3140100993/article/details/105387639
今日推荐