Python如何写入和读取csv文件

写入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