Python对csv文件的读写操作

python内置了csv模块,用它可以方便的操作csv文件。

1、写文件

(1)写文件的方法一

import csv

# open 打开文件有多种模式,下面是常见的4种
# r:读数据,默认模式
# w:写数据,如果已有数据则会先清空
# a:向文件末尾追加数据
# x : 写数据,如果文件已存在则失败
# 第2至4种模式如果第一个参数指定的文件不存在,则会先创建一个空文件
with open('1.csv', 'w', newline='') as f:    
    head = ['标题列1', '标题列2']
    rows = [
                ['张三', 80],
                ['李四', 90]
            ]  
    writer = csv.writer(f) 
    #写入一行数据
    writer.writerow(head) 
    #写入多行数据
    writer.writerows(rows)  

Python对csv文件的读写操作

(2)写文件的方法二

import csv

with open('1.csv', 'w', newline='') as f:    
    head = ['标题列1', '标题列2']
    rows = [
                {'标题列1': '张三', '标题列2' :80},
                {'标题列1': '李四', '标题列2' :90}
            ]
    writer = csv.DictWriter(f,head)
    writer.writeheader()
    writer.writerows(rows)

2、读文件

以上面创建的1.csv为例读文件

import csv

with open('1.csv') as f:    
    reader = csv.reader(f)
    for row in reader:
        print(row)

运行结果:

['标题列1', '标题列2']
['张三', '80']
['李四', '90']