Pandas read_csv add header names以防更改列数

Question

我有很多csv文件，我想与熊猫一起阅读(pd.read_csv文件)，但是，在某些文件中，添加了一个没有标题的列，如以下示例：

Apples, Pears
1, 2
3, 4
5, 6, 7

如果使用pd.read_csv文件（示例_文件）引发以下错误“ParserError:error tokenizeing data。C错误：第4行应有2个字段，saw 3“

我希望避免跳过行，而只需添加一个伪头名称，如Unknown1，并得到以下结果：

^{pr2}$

Answer 1

pandas需要预先知道几何结构才能构建数据帧。您可以读取标题行并添加几个虚拟列名以提供列数，然后重新读取整个csv并丢弃根本没有使用的列。在

>>> import pandas as pd
>>> names = list(pd.read_csv('foo.csv', nrows=0)) + ['unknown1', 'unknown2']
>>> df=pd.read_csv('foo.csv', names=names, skiprows=1).dropna(axis='columns', how='all')
>>> df
   Apples   Pears  unknown1
0       1       2       NaN
1       3       4       NaN
2       5       6       7.0

如果有许多额外的列并且您担心中间数据帧的内存占用，可以使用csv模块扫描文件并计算最大行数。与pandas不同，csv很乐意发出大小不同的行。在

^{pr2}$