Python 创建csv文件并写入数据

1. 什么是CSV文件？

CSV（Comma Separated Values，逗号分隔值）是一种常见的电子表格文件格式，用于将结构化数据存储为纯文本文件。每行代表一个数据记录，每个字段由逗号分隔。CSV文件可以使用任何文本编辑器打开，并且易于与其他程序进行交互。

CSV文件的优势在于它的简单性和广泛的支持。它可以用于存储表格数据、数据库导出、数据交换等多种用途。

2. Python中的csv模块

Python中的csv模块提供了一种简单的方式来读取、写入和操作CSV文件。使用csv模块，我们可以轻松地处理CSV文件中的数据。

首先，我们需要在Python代码中导入csv模块：

import csv

现在，我们可以使用csv模块的函数来处理CSV文件。

3. 创建CSV文件并写入数据

要创建一个新的CSV文件并写入数据，我们首先需要打开一个文件并获得一个文件对象。在这个文件对象上，我们可以使用csv模块的函数来写入数据。

下面是一个示例代码，演示了如何创建一个名为data.csv的CSV文件，并将一些数据写入其中：

import csv

# 要写入CSV文件的数据
data = [
    ['Name', 'Age', 'Country'],
    ['Alice', 25, 'USA'],
    ['Bob', 30, 'Canada'],
    ['Charlie', 35, 'UK']
]

# 打开一个CSV文件并获取文件对象
with open('data.csv', 'w', newline='') as file:

    # 创建一个写入器对象
    writer = csv.writer(file)

    # 写入数据
    writer.writerows(data)

print("数据已成功写入CSV文件。")

运行上述代码后，会创建一个名为data.csv的文件，并将数据写入其中。文件内容如下：

Name, Age, Country
Alice, 25, USA
Bob, 30, Canada
Charlie, 35, UK

注意，我们在open函数的参数中传入了newline=''。这是为了确保在写入CSV文件时不会出现额外的换行符。

4. 追加数据到已存在的CSV文件

如果要将数据追加到已存在的CSV文件中，而不是创建一个新的文件，可以使用open函数的mode参数为'a'（表示追加）。

下面是一个示例代码，演示了如何将数据追加到名为data.csv的已存在的CSV文件中：

import csv

# 要追加到CSV文件的数据
data = [
    ['Diana', 40, 'Brazil'],
    ['Edward', 45, 'Australia']
]

# 打开一个已存在的CSV文件并获取文件对象
with open('data.csv', 'a', newline='') as file:

    # 创建一个写入器对象
    writer = csv.writer(file)

    # 追加数据
    writer.writerows(data)

print("数据已成功追加到CSV文件。")

运行上述代码后，会将新的数据追加到已存在的data.csv文件中。文件内容如下：

Name, Age, Country
Alice, 25, USA
Bob, 30, Canada
Charlie, 35, UK
Diana, 40, Brazil
Edward, 45, Australia

5. 读取CSV文件中的数据

要读取CSV文件中的数据，我们可以使用csv模块的函数。其中，reader函数用于创建一个读取器对象，readrows函数用于逐行读取数据。

下面是一个示例代码，演示了如何读取名为data.csv的CSV文件中的数据：

import csv

# 打开一个CSV文件并获取文件对象
with open('data.csv', 'r') as file:

    # 创建一个读取器对象
    reader = csv.reader(file)

    # 逐行读取并打印数据
    for row in reader:
        print(row)

运行上述代码后，会逐行读取data.csv文件中的数据，并将其打印出来。输出结果如下：

['Name', ' Age', ' Country']
['Alice', ' 25', ' USA']
['Bob', ' 30', ' Canada']
['Charlie', ' 35', ' UK']
['Diana', ' 40', ' Brazil']
['Edward', ' 45', ' Australia']

6. 使用字典读取CSV文件中的数据

除了使用csv.reader函数，我们还可以使用csv.DictReader函数来读取CSV文件中的数据。使用DictReader函数，我们可以将每一行的数据作为一个字典来处理，其中键是列名，值是对应的数据。

下面是一个示例代码，演示了如何使用csv.DictReader函数来读取名为data.csv的CSV文件中的数据：

import csv

# 打开一个CSV文件并获取文件对象
with open('data.csv', 'r') as file:

    # 创建一个字典读取器对象
    reader = csv.DictReader(file)

    # 逐行读取并打印数据
    for row in reader:
        print(row)

运行上述代码后，会逐行读取data.csv文件中的数据，并将其以字典的形式打印出来。输出结果如下：

{'Name': 'Alice', ' Age': ' 25', ' Country': ' USA'}
{'Name': 'Bob', ' Age': ' 30', ' Country': ' Canada'}
{'Name': 'Charlie', ' Age': ' 35', ' Country': ' UK'}
{'Name': 'Diana', ' Age': ' 40', ' Country': ' Brazil'}
{'Name': 'Edward', ' Age': ' 45', ' Country': ' Australia'}

注意，使用DictReader函数时，第一行被假定为列名，并被用作字典的键。