CSV文件的读写-向日葵屋

CSV文件的读写

CSV是一种以半角字符 , 分隔的文本文件。

①普通文本方式读写
可以用普通的读取txt的方式读取，读取结果都是字符串形式。每行内以逗号分隔

f = open('test.csv','r')
lines = f.readlines()
for line in lines:
print(line)
f.close()

写入时按照字符串写入，以逗号分隔，并加换行符\n

f = open('test.csv','w')
f.write('时间,生命\n')
f.write('风景,如画\n')
f.close()

优点快速简洁，缺点行标签，列标签都要自己读写。

②用csv读取和写入
读取：

import csv
with open("test.csv","r") as csvfile:#也可以用open
    reader = csv.reader(csvfile)
    #这里不需要readlines
    for line in reader:
        print line

简单写入用csv库：

import csv
csv_file = open('test.csv','w')
writer = csv.writer(csv_file)
#单行写入用writerow
writer.writerow(['品牌','型号','价值'])
#多行写入用writerows，写入的结果数字往右靠了，字符串往左靠
writer.writerows([['华为','v6 plus',2499],['iphone',7,4500],['samsun','s7',5000]])
csv_file.close()

3、pandas:
读取：

import pandas as pd
data = pd.read_csv('test.csv',encoding ='gbk')
print(data)#打印所有数据
#print(data.head(1))#打印前2行
print(data.columns)#打印所有列名
#print(data.loc[1:2])#打印1到2行
#还有许多许多的用法看收藏链接

用pandas写入：

import pandas as pd
'''pandas学习创建dataframe，然后保存到CSV中，包括列表写入和字典写入：'''
#========================================================
##列表创建dataframe然后保存
#import pandas as pd
#
# 
#a =[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]]
# lindex =['q','w','e','r']
# lcolumns =[6,6,6,6]
# df = pd.DataFrame(a,index = lindex,columns = lcolumns)
# #index为行标签，columns为列标签，不写或者False，那么为0开始的序列
# print(df)
#=============================================================================
#字典创建dateframe然后保存
#字典的每个key的value代表一列，而key是这一列的列名
import pandas as pd
#任意的多组列表
dic ={
    'a_name':[1,2,'',3],'b_name':[4,5,6,6]}    
lindex =['q','w','e','r']
#字典中的key值即为csv中列名
df = pd.DataFrame(dic,index = lindex)
print(df)
#将DataFrame存储为csv,index表示是否显示行名，default=True
df.to_csv("test.csv",index =True)