2022年 11月 3日

【Python】Python读取CSV文件

1. CSV格式

csv 文件格式的本质是一种以文本存储的表格数据(使用 Excel 工具即可读写 csv 文件)。csv 文件的每行代表一行数据,每行数据中每个单元格内的数据以逗号隔开。

Python 提供了 csv 模块来读写 csv 文件。由于 csv 文件的格式本身比较简单(通常第一行是表头,用于说明每列数据的含义,接下来每行代表一行数据),因此使用 csv 模块读取 csv 文件也非常简单:

创建 csv 模块的读取器。

循环调用 csv 读取器的 next() 方法逐行读取 csv 文件内容即可。

next() 方法返回一个 list 列表代表一行数据,list 列表的每个元素代表一个单元格数据。

2.实例代码

  1. # readCSV.py
  2. # python 3.8
  3. import csv
  4. file = 'D:\\1.csv'
  5. with open(file) as f:
  6. reader = csv.reader(f)
  7. header_row = next(reader)
  8. print(header_row)
  9. first_row = next(reader)
  10. print(first_row)

数据如下图:

Excel打开CSV
用Excel打开CSV文件
用记事本打开CSV文件
用记事本打开CSV文件标题
代码输出结果
输出标题和第一行数据

 

3. 更精细的操作

1. 更复杂的读写控制(如筛选行和列),请使用Python第三方库:pandas。详细参考此文:用python读写和处理csv文件。

2. 使用字典方式获取数据,可使用 csv.DictReader(file) ,具体参考:用Python读取CSV文件的5种方式。