Python 导入 CSV 文件

CSV（逗号分隔值）是一种常用的数据格式，用于将数据从一种程序或系统导入到另一种程序或系统中。在Python中，使用适当的库可以轻松地读取和写入CSV文件。

本文将介绍如何使用Python读取CSV文件。我们将使用Python标准库的csv模块来处理CSV文件。

准备工作

在开始编写代码之前，需要准备一个CSV文件。以下是示例CSV文件的示例数据：

姓名, 年龄, 性别
张三, 28, 男
李四, 36, 男
王五, 24, 女

请注意，每个单元格之间都使用逗号分隔。

读取 CSV 文件

要读取CSV文件，我们将使用csv模块的reader函数。以下是一个读取CSV文件并将其打印到控制台的示例代码：

import csv

with open('example.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    line_count = 0
    for row in csv_reader:
        if line_count == 0:
            print(f'列名是: {", ".join(row)}')
            line_count += 1
        else:
            print(f'\t{row[0]}是{row[2]}，今年{row[1]}岁。')
            line_count += 1
    print(f'行数: {line_count}.')

运行此代码将输出以下内容：

列名是: 姓名, 年龄, 性别
        张三是男，今年28岁。
        李四是男，今年36岁。
        王五是女，今年24岁。
行数: 4.

请注意，我们在with语句中打开CSV文件，以确保在读取完文件后自动关闭文件。另外，我们在csv.reader中指定分隔符为逗号。

在for循环中，我们对CSV文件中的每一行进行迭代。如果我们在第一行，我们打印出列名。如果我们不在第一行，我们打印出每一行的数据。

写入 CSV 文件

要写入CSV文件，我们将使用csv模块的writer函数。以下是一个将数据写入CSV文件的示例代码：

import csv

with open('example.csv', mode='w') as csv_file:
    fieldnames = ['姓名', '年龄', '性别']
    writer = csv.DictWriter(csv_file, fieldnames=fieldnames)

    writer.writeheader()
    writer.writerow({'姓名': '张三', '年龄': '28', '性别': '男'})
    writer.writerow({'姓名': '李四', '年龄': '36', '性别': '男'})
    writer.writerow({'姓名': '王五', '年龄': '24', '性别': '女'})

运行此代码将创建一个名为example.csv的新文件，并将以下数据写入其中：

姓名, 年龄, 性别
张三, 28, 男
李四, 36, 男
王五, 24, 女

请注意，我们在open函数中指定了文件模式为’w’，这意味着我们将覆盖现有文件（如果存在）或创建新文件（如果不存在）。我们还使用csv.DictWriter将数据格式化为CSV格式。

结论

Python中的csv模块使读取和写入CSV文件变得非常容易。使用csv.reader函数可以轻松地读取CSV文件，而使用csv.writer函数可以轻松地将数据写入CSV文件。使用这些功能，您可以轻松地处理大量数据，或者将数据从一个程序或系统中导入到另一个程序或系统中。