python筆記-1

（1）寫圖片路徑到txt中小栗子：

# -*- coding: utf-8 -*-

import  glob
a=glob.glob('layers/*.jpg')

for i in range(len(a)):
a[i]=a[i].split('/')[-1]

f=file('list.txt','w')
for i in a:
    f.write(i+'\n')

f.close()

（2）提取圖片路徑中的標籤小栗子：

# -*- coding: utf-8 -*-

with open("/home/sll2/Desktop/train_s.txt","r")as f:
    lines=f.readlines()
with open("/home/sll2/Desktop/train_s.txt","w+")as f_w:
    for line in lines:
        label=line.split('/')[-3]
        #label=int(label)
        line=line.strip('\n')
        f_w.write(line+' '+label+'\n')

f.close()
f_w.close()