在 Python 中,我们常常会遇到需要将运行结果以 CSV 格式导出以供其他语言或工具使用的情况。本文将介绍如何使用 Python 将结果导出为 CSV 格式的两种主要方法。
方法一:使用 csv 模块
csv 模块是 Python 自带的用于读写 CSV 文件的模块。我们可以这样使用它导出 CSV:
import csvwith 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,col3a,b,c 1,2,3
方法二:使用 pandas 库
如果我们的数据是以 pandas DataFrame 的形式存在的,那么可以直接使用 .to_csv() 方法导出 CSV:
import pandas as pddf = 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 格式导出,以供其他语言或工具使用。
来源地址:https://blog.csdn.net/devid008/article/details/130300515