Python - 简单方便的读取CSV文件
Posted SpikeKing
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Python - 简单方便的读取CSV文件相关的知识，希望对你有一定的参考价值。
本文提供一个简易读取CSV的函数，方便后续处理。
- 使用pandas库，读取csv文件
- 获取csv的列名
- 遍历csv每一行，组成字典
- 输出读取进度，避免长时间等待。
- 返回列名，和每一行的字典
源码如下：
def read_csv_file(data_file, num=-1):
"""
读取CSV文件
"""
import pandas
df = pandas.read_csv(data_file)
row_list = []
column_names = list(df.columns)
for idx, row in df.iterrows():
if idx == num:
break
row_list.append(dict(row))
if idx != 0 and idx % 20000 == 0:
print('[Info] idx: {}'.format(idx))
return column_names, row_list
以上是关于Python - 简单方便的读取CSV文件的主要内容，如果未能解决你的问题，请参考以下文章