写入csv文件
import csv
#打开要写入的csv文件,如果该路径下没有此文件,会自动新建一个test.csv文件
f = open('/mnt/sdb/test.csv','w')
csv_writer = csv.writer(f)
#写入列名
csv_writer.writerow(["Name","Label"])
#写入数据(如果是写入多行可以用一个for循环
#前提是[image_name,labels]中只是单行数据,例如[0,4])
csv_writer.writerow([image_name,labels])
'''
Name Label
0 4
'''
#如果[image_name,labels]中存有多行数据,用writerows写入多行数据
#writer.writerows([[0,1,2],[4,5,6]])
'''
Name Label
0 4
1 5
2 6
'''
#关闭文件
f.close()
读取csv文件
采用pandas
import pandas as pd
fpath = "/mnt/sdb/test.csv"
df = pd.read_csv(fpath)
df = df[[ 'Name','Label']]
采用csv
import csv
with open("test.csv","r") as csvfile:
reader = csv.reader(csvfile)
for line in reader:
print line