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