如何使用Python读取CSV文件的标题列?
发布于 2021-01-29 17:19:47
我正在寻找一种方法来仅读取大量大型CSV文件的标题行。
对于每个csv文件,使用Pandas可以使用以下方法:
>>> df = pd.read_csv(PATH_TO_CSV)
>>> df.columns
我可以使用csv模块来做到这一点:
>>> reader = csv.DictReader(open(PATH_TO_CSV))
>>> reader.fieldnames
这些问题是每个CSV文件的大小都超过500MB,仅读取标题行而读取每个文件的整个文件似乎是巨大的浪费。
我所有这些的最终目标是提取唯一的列名。一旦有了每个文件中的列标题列表,便可以执行此操作。
如何快速仅提取CSV文件的标题行?
关注者
0
被浏览
177
1 个回答