python操作csv,現在很多都用pandas包了,不過python還是有一個原始的包可以直接操作csv,或者excel的,下面舉個例子說明csv讀寫csv文件的方法:
import os import csv from PIL import Image import matplotlib.pyplot as plt import time def generate_csv(): path = r"F:\工作文件\達芬奇項目\image_82_83
#這里open一個路徑,向這個文件中寫入數據 with open('../../data/captcha/cnn/lables.csv','w', newline='') as csvfile: svwriter = csv.writer(csvfile, dialect=("excel"))
#這里是csv或者excel的列名 svwriter.writerow(['path']) for maindir, subdir, file_name_list in os.walk(path): for filename in file_name_list: apath = filename # 合並成一個完整路徑
#這里就是插入數據了,一般就是根據自己的情況,我這里是循環一個文件夾,之后把文件夾中的文件名插入csv中 svwriter.writerow([apath]) def read_csv(): path = r"F:\工作文件\達芬奇項目\image_82_83" reader = csv.reader(open('../../data/captcha/cnn/lables.csv','r')) for line in reader:
#這里的line就是讀取的csv中的一行信息,是一個列表,直接可以根據下標來取第幾列
#我這里是把這個路徑的圖片展示出來,不用管下面的信息,只要知道line的類型就可以了 if line[0] != 'path': image_file = path + '\%s' % line[0] print(image_file) im = Image.open(image_file) # im.show() plt.imshow(im) plt.pause(8) plt.close() # time.sleep(5) if __name__ == '__main__': # generate_excel() read_csv()