在Python中读取和写入CSV文件

CSV(Comma Separated Values，逗号分隔值)是一种常见的数据格式，在常规的电子表格和数据库应用中都得到广泛应用。Python内置的csv模块允许我们轻松读取和写入CSV文件。在本文中，我们将介绍如何使用Python操作CSV文件。

读取CSV文件

要读取CSV文件，我们需要首先导入csv模块。然后使用csv.reader函数，打开文件。例如，我们将打开一个名为“example.csv”的文件，如下所示：

import csv

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

我们使用with语句打开文件，并将文件对象传递给csv.reader函数来创建reader对象。csv.reader对象每次呈现一行CSV数据，我们可以通过迭代它来访问文件中的每一行。我们可以使用for循环来迭代整个reader对象并打印每一行。

对于以下CSV文件：

Name,Age,Country
John,28,USA
Emily,22,Canada

以上Python代码将输出：

['Name', 'Age', 'Country']
['John', '28', 'USA']
['Emily', '22', 'Canada']

注意第一行是标题行。

当我们读取CSV文件时，csv.reader将每一行都作为列表返回，其中列表的每个元素都是该行的单元格。

写入CSV文件

要写入CSV文件，我们需要创建一个写入器(writer)对象并将数据写入到该对象中。使用csv.writer函数，我们可以将数据写入CSV文件。

例如，以下代码将创建一个名为“example.csv”的新文件并将数据写入该文件。

import csv

data = [["Name", "Age", "Country"], ["John", "28", "USA"], ["Emily", "22", "Canada"]]

with open('example.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)

在这个例子中，我们使用一个名为“数据”的列表来保存要写入CSV文件的数据。要创建CSV写入器对象，我们使用了csv.writer函数，并传递了一个文件对象和一个空字符串('')作为参数。空字符串（'newline =''）是必需的，以避免写入时出现不必要的换行符。要将数据写入文件，我们使用for循环，逐行遍历数据，并使用csv.writerow函数将每行写入文件。

当我们运行以上Python代码后，将创建一个名为“example.csv”的新文件。该文件将与以下内容填充：

Name,Age,Country
John,28,USA
Emily,22,Canada

以上就是在Python中读写CSV文件的基础。对于更高级的应用程序，csv模块还提供了更多的选项和功能以满足各种需求。