Python 操作 csv文件所有操作汇总大家收藏呀

Python库：csv
Python中集成了专用于处理csv文件的库，名为：csv。

csv 库中有4个常用的对象：

csv.reader：以列表的形式返回读取的数据。
csv.writer：以列表的形式写入数据。
csv.DictReader：以字典的形式返回读取的数据。
csv.DictWriter：以字典的形式写入数据。
读取csv文件
假设现在要读取的csv文件内容如下：

可以看到，该文件的第一行表明数据类型，我们暂且称之为header。从第2行开始，保存的是设备采集到的数据。

使用csv.reader读取数据

导入 csv 库

import csv

以读方式打开文件

with open(“data.csv”, mode=“r”, encoding=“utf-8-sig”) as f:

# 导入 csv 库
import csv

# 以读方式打开文件
with open("data.csv", mode="r", encoding="utf-8-sig") as f:
    
    # 基于打开的文件，创建csv.reader实例
    reader = csv.reader(f)

    # 获取第一行的header
    # header[0] = "设备编号"
    # header[1] = "温度"
    # header[2] = "湿度"
    # header[3] = "转速"
    header = next(reader)

    # 逐行获取数据，并输出
    for row in reader:
        print("{}{}: {}={}, {}={}, {}={}".format(header[0], row[0],
                                                 header[1], row[1],
                                                 header[2], row[2],
                                                 header[3], row[3]))

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

程序运行结果如下：

设备编号0: 温度=31, 湿度=20, 转速=1000
设备编号1: 温度=30, 湿度=22, 转速=998
设备编号2: 温度=32, 湿度=23, 转速=1005
1
2
3

使用csv.DictReader读取数据

# 导入 csv 库
import csv

# 打开文件
with open("data.csv", encoding="utf-8-sig", mode="r") as f:

    # 基于打开的文件，创建csv.DictReader实例
    reader = csv.DictReader(f)

    # 输出信息
    for row in reader:
        print("设备编号{}: 温度={}, 湿度={}, 转速={}".format(row["设备编号"],
                                                   row["温度"],
                                                   row["湿度"],
                                                   row["转速"]))
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

程序运行结果如下：

设备编号0: 温度=31, 湿度=20, 转速=1000
设备编号1: 温度=30, 湿度=22, 转速=998
设备编号2: 温度=32, 湿度=23, 转速=1005
1
2
3

写入csv文件

假设我们现在要创建一个csv文件，将数据保存为如下形式：

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-s0QOqQsT-1667479568458)(C:\Users\Administrator\AppData\Roaming\Typora\typora-user-images\image-20221103203737832.png)]

使用csv.writer写入数据

# 导入 csv 库
import csv

# 创建列表，保存header内容
header_list = ["设备编号", "温度", "湿度", "转速"]

# 创建列表，保存数据
data_list = [
    [0, 31, 20, 1000],
    [1, 30, 22, 998],
    [2, 32, 33, 1005]
]

# 以写方式打开文件。注意添加 newline=""，否则会在两行数据之间都插入一行空白。
with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f:
    
    # 基于打开的文件，创建 csv.writer 实例
    writer = csv.writer(f)

    # 写入 header。
    # writerow() 一次只能写入一行。
    writer.writerow(header_list)

    # 写入数据。
    # writerows() 一次写入多行。
    writer.writerows(data_list)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

使用csv.DictWriter写入数据

# 导入 csv 库
import csv

# 创建 header 列表
header_list = ["设备编号", "温度", "湿度", "转速"]

# 创建数据列表，列表的每个元素都是字典
data_list = [
    {"设备编号": "0", "温度": 31, "湿度": 20, "转速": 1000},
    {"设备编号": "1", "温度": 30, "湿度": 22, "转速": 998},
    {"设备编号": "2", "温度": 32, "湿度": 23, "转速": 1005},
]

# 以写方式打开文件。注意添加 newline=""，否则会在两行数据之间都插入一行空白。
with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f:
    
    # 基于打开的文件，创建 csv.DictWriter 实例，将 header 列表作为参数传入。
    writer = csv.DictWriter(f, header_list)

    # 写入 header
    writer.writeheader()

    # 写入数据
    writer.writerows(data_list)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

关于写入，需要注意：

• 在打开文件时，需要添加newline = ""。否则，会在每2行有效内容之间添加一行空白。
• 如果要保存的内容有中文，而且之后需要用Excel打开文件，那么需要选用utf-8-sig编码。如果使用utf-8编码，会导致使用Excel查看文件时中文乱码。