Python代码编写案例：如何读取和写入CSV文件

读取和写入CSV（逗号分隔值）文件在Python中是非常常见的任务，它们是一种常用的数据交换格式。CSV文件可以由电子表格软件生成，并且易于使用文本编辑器编辑。Python中有许多库可以帮助我们读取和写入CSV文件，如内置的csv模块、pandas库等。在本文中，我将介绍如何使用csv模块来读取和写入CSV文件，并提供一些使用示例。

首先，让我们看一下如何读取CSV文件。首先，我们需要使用open()函数打开CSV文件，并指定文件的路径和文件模式。文件模式可以是'r'表示只读模式。然后，我们可以使用csv模块的reader()函数创建一个CSV读取器对象，并将文件对象作为参数传递给它。最后，我们可以使用for循环遍历读取器对象，并将每行数据打印出来。

下面是一个读取CSV文件的示例代码：

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建读取器对象
    reader = csv.reader(file)
    
    # 遍历读取器对象
    for row in reader:
        # 打印每行数据
        print(row)

上述代码假设CSV文件名为data.csv，位于当前工作目录中。在使用with语句打开文件后，我们可以使用reader对象迭代获取文件中的每行数据，每行数据以列表形式表示。

下面是一个示例CSV文件的内容：

Name,Age,City
John,25,New York
Jane,30,Los Angeles

运行上述示例代码后，输出将是：

['Name', 'Age', 'City']
['John', '25', 'New York']
['Jane', '30', 'Los Angeles']

接下来，让我们看一下如何写入CSV文件。同样地，我们需要使用open()函数打开CSV文件，并指定文件的路径和文件模式。文件模式可以是'w'表示写入模式。然后，我们可以使用csv模块的writer()函数创建一个CSV写入器对象，并将文件对象作为参数传递给它。最后，我们可以使用writerow()方法将列表作为参数传递给写入器对象，以将数据写入CSV文件。

下面是一个写入CSV文件的示例代码：

import csv

# 准备数据
data = [
    ['Name', 'Age', 'City'],
    ['John', '25', 'New York'],
    ['Jane', '30', 'Los Angeles']
]

# 打开CSV文件
with open('output.csv', 'w', newline='') as file:
    # 创建写入器对象
    writer = csv.writer(file)
    
    # 写入数据
    for row in data:
        writer.writerow(row)

上述代码创建了一个名为output.csv的CSV文件，并将数据写入其中。注意要使用newline=''参数来避免在写入文件时添加额外的换行符。

运行上述示例代码后，将创建一个名为output.csv的CSV文件，其中包含了我们准备的数据。

总结：

读取和写入CSV文件在Python中非常简单。通过使用csv模块，我们可以轻松地读取和写入CSV文件，处理各种数据集。在本文中，我们介绍了如何使用csv模块来读取和写入CSV文件，并提供了一些使用示例。希望这些示例对您有所帮助！