在Python中进行CSV文件的读取与写入操作_Python

1 引言

在数据处理和数据分析领域，csv (逗号分隔值) 文件是一种常见的文件格式，用于存储表格数据。python 通过内置的 csv 模块提供了对 csv 文件的读写支持，使得处理这种类型的文件变得简单高效。本文将详细介绍如何在 python 中进行 csv 文件的读取和写入操作。

2 csv 文件简介

csv 文件是一种简单的文本文件，其中的数据由逗号分隔。每行代表一个数据记录，每个记录可以包含多个字段。csv 文件被广泛用于数据交换和数据存储。

3 python 中的 csv 模块

python 的 csv 模块提供了一系列函数来读取和写入 csv 文件。它可以处理各种类型的 csv 数据，包括不同的分隔符和引号约定。

4 写入 csv 文件

4.1 基本用法

import csv

# 数据
data = [
    ['name', 'age', 'city'],
    ['alice', 30, 'new york'],
    ['bob', 25, 'los angeles']
]

# 写入 csv 文件
with open('example.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

4.2 高级用法

使用 dictwriter 写入 csv，提供了字段名的映射。

with open('example.csv', 'w', newline='') as file:
    fields = ['name', 'age', 'city']
    writer = csv.dictwriter(file, fieldnames=fields)

    writer.writeheader()
    writer.writerow({'name': 'alice', 'age': 30, 'city': 'new york'})
    writer.writerow({'name': 'bob', 'age': 25, 'city': 'los angeles'})

5 读取 csv 文件

5.1 基本用法

# 读取 csv 文件
with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

5.2 高级用法

使用 dictreader 读取 csv，每行数据被映射到一个字典。

with open('example.csv', 'r') as file:
    reader = csv.dictreader(file)
    for row in reader:
        print(row)

6 实例演示

以下是一个 python 示例，演示如何使用 csv 模块来读取和写入 csv 文件。这个示例将包含两个主要部分：一个用于写入 csv 文件的函数和另一个用于读取 csv 文件的函数。我们将这些功能封装在一个名为 csv_example.py 的 python 文件中。

# csv_example.py
import csv

def write_csv(file_name, data, field_names=none):
    """ 写入 csv 文件 """
    with open(file_name, 'w', newline='', encoding='utf-8') as file:
        if field_names:
            writer = csv.dictwriter(file, fieldnames=field_names)
            writer.writeheader()
            writer.writerows(data)
        else:
            writer = csv.writer(file)
            writer.writerows(data)

def read_csv(file_name):
    """ 读取 csv 文件 """
    with open(file_name, 'r', newline='', encoding='utf-8') as file:
        reader = csv.reader(file)
        for row in reader:
            print(row)

def main():
    # csv 文件名
    file_name = 'example.csv'

    # 数据 - 字典格式
    data_dict = [
        {'name': 'alice', 'age': 30, 'city': 'new york'},
        {'name': 'bob', 'age': 25, 'city': 'los angeles'}
    ]

    # 数据 - 列表格式
    data_list = [
        ['name', 'age', 'city'],
        ['alice', 30, 'new york'],
        ['bob', 25, 'los angeles']
    ]

    # 写入 csv 文件
    write_csv(file_name, data_dict, field_names=['name', 'age', 'city'])
    # 或者使用列表数据
    # write_csv(file_name, data_list)

    # 读取 csv 文件
    read_csv(file_name)

if __name__ == '__main__':
    main()

运行以上代码控制台输出结果：[‘name’, ‘age’, ‘city’]
[‘alice’, ‘30’, ‘new york’]
[‘bob’, ‘25’, ‘los angeles’]
并生成文件：“example.csv”，内容如下：

这个脚本定义了两个函数：write_csv 和 read_csv。write_csv 函数可以根据传入的数据格式（字典或列表）将数据写入 csv 文件。read_csv 函数则用于读取 csv 文件并打印其内容。main 函数中整合了这些操作的流程。运行这个脚本将会创建一个名为 example.csv 的文件，并在其中写入数据，然后读取并打印出这些数据。