在 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 格式导出,以供其他语言或工具使用。
文章出处登录后可见!
已经登录?立即刷新