python3使用csv包，读写csv文件

时间：2024-01-10 08:27:32

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()

相关文章