python对csv文件的读写操作

Posted 2022-08-23 gdjlc

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)  

（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‘]