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

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

方法一:使用 csv 模块

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

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

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

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

方法二:使用 pandas 库

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

1.  
   import pandas as pd
2.  
    
3.  
   df = pd.DataFrame({
4.  
       'col1': ['a', 'b', 'c'],
5.  
       'col2': [1, 2, 3], 
6.  
       'col3': [1.5, 2.5, 3.5]
7.  
   })
8.  
    
9.  
   df.to_csv('data.csv')

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

这篇好文章是转载于：学新通技术网