Python使用csv模块读写csv文件

csv是逗号分隔值（Comma-Separated Values）的简称。

有时也称为字符分隔值，因为分隔字符也可以不是逗号，其文件以纯文本形式存储表格数据。纯文本意味着该文件是一个字符序列，不含必须像二进制数字那样被解读的数据。

csv可以存储各种各样的数据，不过，通常来说，比较适合存储有相同字段或表头的一批数据，这些数据可以展示成表格。

可以使用excel开启csv文件，打开后看到的数据以excel表格的方式进行展示。

现在我们就开始使用csv将数据写入csv文件，然后将数据从csv中读取出来使用。

一、将数据写入csv文件中

1. import csv
2. 

3. 

4. csv_data = (
5. (1, 2, 3, 4, 5, 6),
6. ('a', 'b', 'c', 'd', 'e', 'f'),
7. ('p', 'y', 't', 'h', 'o', 'n')
8. )
9. output_file_name = 'csv_file.csv'
10. 

11. 

12. def save_csv(target_list, output_file_name):
13. """
14. 将数据写入csv文件
15. """
16. if not output_file_name.endswith('.csv'):
17. output_file_name += '.csv'
18. csv_file = open(output_file_name, "w", newline="")
19. key_data = target_list[0]
20. value_data = [target for target in target_list]
21. csv_writer = csv.writer(csv_file)
22. csv_writer.writerow(key_data)
23. csv_writer.writerows(value_data)
24. csv_file.close()
25. 

26. 

27. save_csv(csv_data, output_file_name)

代码描述：

1.我们先将需要保存的数据解析好，保存成固定的数据类型（保存成列表，元组，字典都可以，根据具体场景来选择）

2.我们将保存数据到csv文件的代码封装成一个函数，方便重用。步骤主要分为三步：打开文件，写入数据，关闭文件。其中，写入数据时记得先写入表头（我们使用excel打开时需要表头）再写入表格中的数据，数据要以一个列表的形式传递给writerows()。

运行结果：

运行以上代码后，会在当前目录下创建一个csv_file.csv的文件，并写入csv_data的数据，可以使用excel打开文件查看。如下图。

二、从csv文件中读取数据

1. input_file_name = 'csv_file.csv'
2. 

3. 

4. def read_csv(input_file_name):
5. """
6. 读取csv文件数据
7. """
8. with open(input_file_name, 'r', encoding='utf-8') as csv_file:
9. csv_reader = csv.reader(csv_file)
10. # csv_reader对象，是一个列表的格式
11. print(csv_reader)
12. # csv_reader对象的一个迭代器，可以通过next()取出其中的元素
13. print(next(csv_reader))
14. # 也可以通过for循环取出所有元素
15. for line in csv_reader:
16. print(''.join(line))
17. 

18. 

19. read_csv(output_file_name)

代码描述：

1.可以通过with上下文管理的方式打开csv文件，如果想在with的代码块外对读出来的数据进行处理，则可以使用open()打开，再使用close()关闭。

2.csv通过csv.reader()来打开csv文件，返回的是一个列表格式的迭代器，可以通过next()方法获取其中的元素，也可以使用for循环依次取出所有元素。

运行结果：

1. <_csv.reader object at 0x00000295BC044528>
2. ['1', '2', '3', '4', '5', '6']
3. 123456
4. abcdef
5. python

这样，将数据写入csv和从csv中读取数据就完成了，使用过程是非常简单的。