在python中使用csv

2018-12-02 本文已影响1人 dmzlingyin

在python中使用csv

这篇文章讨论的如何使用python加载和解析csv文件

首先，什么是csv呢？

CSV(逗号分隔值)是一种用于存储表格数据(如电子表格或数据库)的简单文件格式。CSV文件以纯文本形式存储表格数据(数字和文本)。文件的每一行都是一个数据记录。每个记录由一个或多个字段组成，字段之间用逗号分隔。逗号作为字段分隔符的使用是该文件格式名称的来源。

由于csv是python的内建模块，所以我们不需要安装它

1、读取csv文件

#首先，导入csv模块
import csv

#被读取的csv文件名
filename = 'example.csv'

#初始化字段和行列表
fields = []
rows = []


#打开并读取
with open(filename,'r') as csvfile:
    #创建一个csv reader对象
    csvreader = csv.reader(csvfile)
    
    #通过第一行，提取csv文件的字段
    #在python3中，需要使用next(csvreader),但是在Python2中，使用csvreader.next()
    #fields = csvreader.next()
    fields = next(csvreader)

    #逐行读取数据
    for row in csvreader:
        rows.append(row)

    #获取总行数
    print('Total no. of rows: %d' % (csvreader.line_num))

#打印字段名称 
print('field names are:' + ','.join(field for field in fields))

#打印前五行
print('\nFist 5 rows are:\n')
for row in rows[:5]:
    #解析行的每一列
    for col in row:
        print('%10s' % col)
    print('\n')

我们来分析一下上面的代码

with open(filename, 'r') as csvfile:
csvreader = csv.reader(csvfile)

在这里，我们首先以读取模式打开CSV文件。文件对象被命名为csvfile。文件对象被转换为csv。我们保存csv为csvreader。

fields = csvreader.next()

csvreader是一个可迭代的对象。因此，.next()方法返回当前行并将迭代器推进到下一行。因为csv文件的第一行包含标题(或字段名)，所以我们将它们保存在一个名为fields的列表中。

for row in csvreader:
    rows.append(row)

现在，我们使用for循环遍历剩余的行。每一行都被添加到一个名为rows的列表中。如果尝试打印每一行，您会发现该行只是一个包含所有字段值的列表。

print("Total no. of rows: %d"%(csvreader.line_num))

csvreader.line_num只是一个计数器，它返回已迭代的行数。

1、写入csv

import csv

#字段名称
fields = ['Name','Branch','Year','CGPA']

#行数据
rows = [ ['Nikhil', 'COE', '2', '9.0'], 
     ['Sanchit', 'COE', '2', '9.1'], 
     ['Aditya', 'IT', '2', '9.3'], 
     ['Sagar', 'SE', '1', '9.5'], 
     ['Prateek', 'MCE', '3', '7.8'], 
     ['Sahil', 'EP', '2', '9.1']] 

#csv文件的名字
filename = 'university_records.csv'

#写入csvw文件
with open(filename,'w') as csvfile:
    csvwriter = csv.writer(csvfile)

    #写入字段
    csvwriter.writerow(fields)
    
    #写入行数据
    csvwriter.writerows(rows)

分析代码

字段和行已经定义。字段是包含所有字段名称的列表。行是列表的列表。每一行都是一个包含该行字段值的列表。

with open(filename, 'w') as csvfile:
csvwriter = csv.writer(csvfile)

在这里，我们首先以写模式打开CSV文件。文件对象被命名为csvfile。文件对象被转换为csv。我们保存csv.writer对象作为csvwriter。

csvwriter.writerow(fields)

现在我们使用writerow方法来写入第一行，这一行就是字段名。

csvwriter.writerows(rows)

我们使用writerows方法一次写多行

3、写csv写入一个字典

import csv 

# 字典对象的行数据
mydict =[{'branch': 'COE', 'cgpa': '9.0', 'name': 'Nikhil', 'year': '2'}, 
     {'branch': 'COE', 'cgpa': '9.1', 'name': 'Sanchit', 'year': '2'}, 
     {'branch': 'IT', 'cgpa': '9.3', 'name': 'Aditya', 'year': '2'}, 
     {'branch': 'SE', 'cgpa': '9.5', 'name': 'Sagar', 'year': '1'}, 
     {'branch': 'MCE', 'cgpa': '7.8', 'name': 'Prateek', 'year': '3'}, 
     {'branch': 'EP', 'cgpa': '9.1', 'name': 'Sahil', 'year': '2'}] 

# 字段名称 
fields = ['name', 'branch', 'year', 'cgpa'] 

# csv文件名称 
filename = "university_records.csv"

# 写入csv 
with open(filename, 'w') as csvfile: 
    # 创建一个csv dict writer 对象
    writer = csv.DictWriter(csvfile, fieldnames = fields) 
  
    #字段的写入 
    writer.writeheader() 
  
    # 行数据的写入 
    writer.writerows(mydict)

这个例子中，我们将mydict写入csv文件

with open(filename, 'w') as csvfile:
writer = csv.DictWriter(csvfile, fieldnames = fields)

在这里，文件对象(csvfile)被转换为一个DictWriter对象。我们将字段名指定为参数。

 writer.writeheader()

writeheader方法仅使用预先指定的字段名写入csv文件的第一行。

writer.writerows(mydict)

writerows方法只写入所有行，但是在每一行中，它只写入值(而不是key)。

在python中使用csv