Python 导入CSV文件详解|极客笔记

Python 导入CSV文件详解

在数据分析和数据处理工作中，CSV（Comma-Separated Values）文件是一种常见的数据格式。CSV文件通常由文本文件组成，其中的数据是以逗号分隔的。在Python中，我们可以使用csv模块来方便地读取和写入CSV文件。本文将详细讲解在Python中如何导入CSV文件。

1. 导入csv模块

在Python中，我们可以通过import关键字导入csv模块，以便使用其中提供的方法来处理CSV文件。以下是导入csv模块的代码示例：

import csv

2. 读取CSV文件数据

在Python中，我们可以使用csv.reader()方法来读取CSV文件中的数据。下面是一个读取CSV文件数据的示例代码：

import csv

# 打开CSV文件
with open('data.csv', mode='r') as file:
    csv_reader = csv.reader(file)

    # 遍历CSV文件中的每一行数据
    for row in csv_reader:
        print(row)

运行以上代码，可以打印出CSV文件中的每一行数据。

3. 写入数据到CSV文件

除了读取数据，我们还可以使用csv.writer()方法将数据写入到CSV文件中。下面是一个将数据写入CSV文件的示例代码：

import csv

# 要写入的数据
data = [
    ['John', 'Doe', 25],
    ['Jane', 'Smith', 30],
    ['Tom', 'Brown', 28]
]

# 打开CSV文件并写入数据
with open('output.csv', mode='w') as file:
    csv_writer = csv.writer(file)

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

print("数据已写入到output.csv文件中。")

运行以上代码，可以将数据写入到output.csv文件中。

4. 使用DictReader和DictWriter

除了reader()和writer()方法外，csv模块还提供了DictReader()和DictWriter()方法，这两个方法允许我们使用字典格式来读取和写入数据。下面是一个使用DictReader()和DictWriter()方法的示例代码：

import csv

# 读取CSV文件数据
with open('data.csv', mode='r') as file:
    csv_reader = csv.DictReader(file)

    # 遍历CSV文件中的每一行数据
    for row in csv_reader:
        print(row)

# 写入数据到CSV文件
data = [
    {'First Name': 'John', 'Last Name': 'Doe', 'Age': 25},
    {'First Name': 'Jane', 'Last Name': 'Smith', 'Age': 30},
    {'First Name': 'Tom', 'Last Name': 'Brown', 'Age': 28}
]

with open('output.csv', mode='w') as file:
    fieldnames = ['First Name', 'Last Name', 'Age']
    csv_writer = csv.DictWriter(file, fieldnames=fieldnames)

    csv_writer.writeheader()

    for row in data:
        csv_writer.writerow(row)

以上代码演示了如何使用DictReader()和DictWriter()方法来读取和写入CSV文件。