1. 程式人生 > >python 從txt檔案查詢含某欄位的行,並將這些行名稱的影象提取出來另存

python 從txt檔案查詢含某欄位的行,並將這些行名稱的影象提取出來另存

初學python,雖然方法笨拙,但是為了便於學習做紀錄。

# -*- coding: UTF-8 -*- 
#!/usr/bin/env python
import re
from PIL import Image
f1 = open('1.txt','r')
f2 = open('car_val1.txt','w+') 
for line in f1.readlines():
if re.findall(' 1',line): #查詢“空格1”的行 每行的格式000005 -1\n 000007  1
        f2.write(line) #把查詢到的行寫入f2.
f1.close()
f2.close()
f3 = open('car_val1.txt','r')
for line2 in f3.readlines():
line3=line2[:-4] #讀取每行去掉後四位的數
im = Image.open('../VOCdevkit/VOC2007/JPEGImages/{}.jpg'.format(line3))#開啟改路徑下的line3記錄的的檔名
im.save('./car/{}.jpg'.format(line3)) #把資料夾中指定的檔名稱的圖片另存到該路徑下
f3.close()