Python导入CSV数据|极客教程

Python导入CSV数据

1. 引言

CSV(Comma Separated Values)是一种常见的文本文件格式，用于存储表格数据。在许多应用程序中，我们经常需要将CSV文件导入到Python中进行数据处理和分析。本文将详细介绍如何使用Python导入CSV数据，并提供示例代码和运行结果。

2. 导入CSV模块

在Python中，我们可以使用内置的csv模块来处理CSV文件。首先，我们需要在代码中导入csv模块：

import csv

3. 读取CSV文件

在Python中，我们可以使用csv.reader()函数来读取CSV文件。该函数将返回一个可迭代的对象，我们可以通过遍历来获取文件中的每一行数据。以下是使用csv.reader()函数读取CSV文件的示例代码：

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    reader = csv.reader(file)

    # 遍历每一行数据并输出
    for row in reader:
        print(row)

上述代码中，我们首先通过open()函数打开了一个名为data.csv的CSV文件，并以只读模式打开('r')。接下来，我们使用csv.reader()函数创建了一个CSV读取器对象reader。然后，我们可以通过遍历reader对象来获取文件中的每一行数据，并通过print()函数输出。

4. 指定分隔符和引号字符

默认情况下，csv.reader()函数将使用逗号(,)作为分隔符，并将双引号(“)作为引号字符。如果CSV文件的分隔符和引号字符与默认值不同，我们可以在调用csv.reader()函数时指定分隔符和引号字符。以下是示例代码：

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    reader = csv.reader(file, delimiter=';', quotechar="'")

    # 遍历每一行数据并输出
    for row in reader:
        print(row)

在上述代码中，我们在调用csv.reader()函数时，通过delimiter参数指定了分隔符为分号(;)，通过quotechar参数指定了引号字符为单引号(‘)。

5. 读取特定列的数据

有时候，我们只需要读取CSV文件中的某些列数据，而不是全部数据。在这种情况下，我们可以使用Python的列表切片操作来筛选特定的列。以下是示例代码：

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    reader = csv.reader(file)

    # 读取第1列和第3列数据
    for row in reader:
        column1 = row[0]
        column3 = row[2]
        print(column1, column3)

上述代码中，我们通过索引来获取第1列和第3列的数据，并分别赋值给变量column1和column3。然后，我们使用print()函数输出这两列数据。

6. 写入CSV文件

除了读取CSV文件，Python还可以将数据写入CSV文件。我们可以使用csv.writer()函数创建一个CSV写入器对象，然后使用writerow()方法将数据写入文件。以下是示例代码：

# 创建要写入的数据
data = [
    ['Name', 'Age', 'Gender'],
    ['Alice', 25, 'Female'],
    ['Bob', 30, 'Male'],
    ['Charlie', 35, 'Male']
]

# 打开CSV文件
with open('output.csv', 'w', newline='') as file:
    # 创建CSV写入器
    writer = csv.writer(file)

    # 写入数据
    for row in data:
        writer.writerow(row)

上述代码中，我们首先创建了一个二维列表data，其中包含了要写入CSV文件的数据。然后，我们通过open()函数打开一个名为output.csv的CSV文件，并以写入模式打开('w')。接下来，我们使用csv.writer()函数创建一个CSV写入器对象writer。最后，我们通过遍历data列表，将每一行数据使用writer.writerow()方法写入CSV文件。