Python 如何将运行结果导出为 CSV 格式?

在 Python 中,我们常常会遇到需要将运行结果以 CSV 格式导出以供其他语言或工具使用的情况。本文将介绍如何使用 Python 将结果导出为 CSV 格式的两种主要方法。

方法一:使用 csv 模块

csv 模块是 Python 自带的用于读写 CSV 文件的模块。我们可以这样使用它导出 CSV:

import csv

with open('data.csv', 'w') as f:
    writer = csv.writer(f)
    writer.writerow(['col1', 'col2', 'col3'])
    writer.writerow(['a', 'b', 'c'])
    writer.writerow(['1', '2', '3'])

这会生成如下 data.csv 文件:

col1,col2,col3
a,b,c  
1,2,3

方法二:使用 pandas 库

如果我们的数据是以 pandas DataFrame 的形式存在的,那么可以直接使用 .to_csv() 方法导出 CSV:

import pandas as pd

df = pd.DataFrame({
    'col1': ['a', 'b', 'c'],
    'col2': [1, 2, 3], 
    'col3': [1.5, 2.5, 3.5]
})

df.to_csv('data.csv')

这同样会生成上面内容的 CSV 文件。
pandas 的 .to_csv() 方法还有很多参数可以定制 CSV 导出的格式,如 header、index、na_rep 等,可以根据需求选择使用。
总之,无论是使用 csv 模块还是 pandas 的 .to_csv() 方法,Python 都可以非常方便地将结果以 CSV 格式导出,以供其他语言或工具使用。