Python使用csv模块读写csv文件

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

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

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

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

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

一、将数据写入csv文件中

import csv


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


def save_csv(target_list, output_file_name):
    """
    将数据写入csv文件
    """
    if not output_file_name.endswith('.csv'):
        output_file_name += '.csv'
    csv_file = open(output_file_name, "w", newline="")
    key_data = target_list[0]
    value_data = [target for target in target_list]
    csv_writer = csv.writer(csv_file)
    csv_writer.writerow(key_data)
    csv_writer.writerows(value_data)
    csv_file.close()


save_csv(csv_data, output_file_name)

代码描述：

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

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

运行结果：

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

二、从csv文件中读取数据

input_file_name = 'csv_file.csv'


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


read_csv(output_file_name)

代码描述：

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

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

运行结果：

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

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