CSV文件的读写

深藏阁楼爱情的钟 2022-02-05 08:53 413阅读 0赞

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