快捷導(dǎo)航

Python對(duì)130w+張圖片檢索的實(shí)現(xiàn)方法

更新時(shí)間：2021年03月12日 10:16:01 作者：aquariusmao

這篇文章主要給大家介紹了關(guān)于Python對(duì)130w+張圖片檢索的實(shí)現(xiàn)方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

任務(wù)說(shuō)明：

130w+張圖片，8張excel表里記錄了需要檢索圖片的文件名，現(xiàn)在需要找出對(duì)應(yīng)的圖片，將找出的圖片按不同的excel分別保存，并且在excel里能夠直接打開(kāi)圖片。

任務(wù)分析：

如果數(shù)據(jù)量不大的話，可以直接讀取excel表里的文件名進(jìn)行搜索保存，但這次的任務(wù)顯然不合適，因?yàn)閳D片實(shí)在太多，所以考慮后按照以下步驟：

1、遍歷圖片文件夾，讀取文件名和文件路徑，寫(xiě)入到csv文件中；

2、使用pandas的merge函數(shù)，實(shí)現(xiàn)8張?jiān)糴xcel表與csv文件根據(jù)圖片文件名的對(duì)碰；

3、使用shutil的copy函數(shù)，讀取文件路徑進(jìn)行保存。

代碼分析：

1、文件遍歷

import os
import pandas as pd
 
file_list = []
path_list = []
path = r"此處添加圖片路徑"
print("任務(wù)開(kāi)始")
for root,dirs,files in os.walk(path):
 for file in files:
  file_list.append(file.split('.')[0])
  path_list.append(os.path.join(root,file))
print("文件遍歷結(jié)束")
file_dic = dict(zip(file_list,path_list))
df = pd.DataFrame.from_dict(file_dic,orient='index', columns=['圖片1路徑']).reset_index().rename(columns={'index':'圖片1'})
df.to_csv("圖片1.csv")
df = pd.DataFrame.from_dict(file_dic,orient='index', columns=['圖片2路徑']).reset_index().rename(columns={'index':'圖片2'})
df.to_csv("圖片2.csv")
df = pd.DataFrame.from_dict(file_dic,orient='index', columns=['圖片3路徑']).reset_index().rename(columns={'index':'圖片3'})
df.to_csv("圖片3.csv")
print("文件目錄導(dǎo)出成功")

2、表格對(duì)碰

import pandas as pd
frame1 = pd.read_excel(r'excel表1.xlsx', 'sheet名')
frame2 = pd.read_csv(r'圖片1.csv', sep=',')
frame3 = pd.read_csv(r'圖片2.csv', sep=',')
frame4 = pd.read_csv(r'圖片3.csv', sep=',')
frame5 = pd.merge(frame1, frame2, on = ['圖片1'], how = 'left')
frame6 = pd.merge(frame5, frame3, on = ['圖片2'], how = 'left')
frame7 = pd.merge(frame6, frame4, on = ['圖片3'], how = 'left')
col = ['圖片1','圖片2','圖片3']
frame7[col] = frame7[col].fillna('未找到')
frame7.to_excel('excel表1合并后.xlsx')

3、圖片復(fù)制

import shutil
target = '此處為excel表1導(dǎo)出圖片路徑'
copylist1 = frame7['圖片1']
for src in copylist1:
  if src != '未找到':
    shutil.copy(src, target)
copylist2 = frame7['圖片2']
for src in copylist2:
  if src != '未找到':
    shutil.copy(src, target)
copylist3 = frame7['圖片3']
for src in copylist3:
  if src != '未找到':
    shutil.copy(src, target)
print('復(fù)制完畢')

4、excel里打開(kāi)圖片，可以使用excel自帶的hyperlink函數(shù)。