Python打开CSV文件详解|极客笔记

Python打开CSV文件详解

CSV（Comma-Separated Values）是一种常见的数据格式，通常用逗号来分隔不同的字段。在Python中，我们可以使用csv模块来打开、读取和写入CSV文件。

1. 打开CSV文件

要打开CSV文件，我们首先需要导入csv模块，然后使用open()函数来打开文件。我们可以指定文件的路径和打开模式（读取、写入、追加等）。

下面是一个简单的示例，演示如何打开一个名为data.csv的CSV文件：

import csv

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

在这个示例中，我们首先导入csv模块，然后使用open()函数打开名为data.csv的CSV文件。我们指定文件打开模式为’r’，即读取模式。然后我们使用csv.reader()函数创建一个reader对象，这个对象可以用来迭代CSV文件的每一行。最后，我们遍历reader对象并打印每一行的内容。

2. 读取CSV文件

一旦我们成功打开了CSV文件，就可以开始读取文件中的数据了。csv.reader对象可以让我们逐行读取文件内容，并将每行分割为一个列表。

下面是一个示例代码，演示如何读取CSV文件中的数据并打印出来：

import csv

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

假设data.csv文件的内容如下：

Name,Age,City
Alice,25,New York
Bob,30,Los Angeles
Charlie,22,Chicago

运行上面的代码，输出将是：

['Name', 'Age', 'City']
['Alice', '25', 'New York']
['Bob', '30', 'Los Angeles']
['Charlie', '22', 'Chicago']

在输出中，每一行的数据都被分割成了一个包含多个元素的列表。我们可以根据列表的索引来访问每个字段的数值。

3. 写入CSV文件

除了读取CSV文件，我们也可以使用csv模块来写入CSV文件。csv.writer对象可以让我们将数据写入CSV文件中。

下面是一个示例代码，演示如何将数据写入一个名为output.csv的CSV文件中：

import csv

data = [
    ['Name', 'Age', 'City'],
    ['David', 28, 'San Francisco'],
    ['Eve', 33, 'Seattle'],
    ['Frank', 19, 'Boston']
]

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

在这个示例中，我们定义了一个包含数据的二维列表data。然后，我们使用open()函数打开名为output.csv的CSV文件，并指定打开模式为’w’，即写入模式。我们还加上了参数newline=”，这样可以避免在写入文件时出现多余的空白行。

接着，我们使用csv.writer()函数创建一个writer对象，这个对象可以让我们将数据写入CSV文件中。最后，使用writer.writerows()方法将整个二维列表data写入CSV文件中。