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