python3使用csv包,读写csv文件

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()
原文地址:https://www.cnblogs.com/gunduzi/p/10725839.html