Python输出CSV|极客教程

Python输出CSV

概述

CSV（Comma-Separated Values）是一种常见的文件格式，用于存储和交换数据。Python作为一种强大的编程语言，在处理CSV文件方面提供了丰富的库和功能。本文将详细介绍如何使用Python输出CSV文件，并给出相关的代码示例以及运行结果。

CSV文件格式

CSV文件由逗号分隔的值组成，每行代表一条记录，每个字段由逗号分隔。通常，CSV文件的第一行会包含字段名，从第二行开始是具体的数据。例如，下面是一个简单的示例CSV文件：

Name,Age,City
John,25,New York
Amy,28,San Francisco

使用CSV模块输出CSV文件

Python的标准库中提供了csv模块，以便我们更方便地进行CSV文件的读写操作。下面是一个使用csv模块输出CSV文件的示例代码：

import csv

data = [
    ['Name', 'Age', 'City'],
    ['John', 25, 'New York'],
    ['Amy', 28, 'San Francisco']
]

filename = 'output.csv'

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

print(f"CSV文件 '{filename}' 输出成功！")

运行结果：

CSV文件 'output.csv' 输出成功！

以上代码通过导入csv模块，创建一个二维列表data来存储数据，并指定要输出的文件名为output.csv。然后，通过open()函数以写入模式打开文件，并使用csv.writer()来创建一个写入器。最后，通过writer.writerows()将数据写入文件。运行结果显示输出成功。

若要在程序运行过程中添加数据，只需将数据追加到data列表中，然后再次调用writer.writerows()即可。

使用pandas库输出CSV文件

除了标准库的csv模块外，还有许多第三方库可供我们使用。其中，pandas库是一个功能强大且广泛使用的数据处理库，它提供了更简单的方式来处理CSV文件。下面是一个使用pandas库输出CSV文件的示例代码：

import pandas as pd

data = {
    'Name': ['John', 'Amy'],
    'Age': [25, 28],
    'City': ['New York', 'San Francisco']
}

df = pd.DataFrame(data)

filename = 'output.csv'
df.to_csv(filename, index=False)

print(f"CSV文件 '{filename}' 输出成功！")

运行结果：