python - 从 csv 读取时如何检测和删除数据集上方的行？

Question

我有一个看起来像这样的 csv:

name: john
date modified: 2018-09
from: jane
colum1 column2 column3
data    data    data 

在从 csv 读取表格数据之前，我可以应用任何函数来去除任何行吗？目前 column 上面的行在我读入时看起来像奇怪的字符。

新表应该是这样的:

colum1 column2 column3
data    data    data 

Answer 1

我会做这样的事情:

from io import StringIO
with open('filename.csv') as f:
    lines = f.readlines()
s = StringIO(''.join((l for l in lines if ':' not in l)))
pd.read_csv(s)

或者:

with open('filename.csv') as f:
    lines = f.readlines()
skip_rows_idx = [i for i, l in enumerate(lines) if ':' in l]
pd.read_csv('filename.csv', skiprows=skip_rows_idx)

如果标题中没有冒号，则可以修改上面的代码(第一个示例)以删除第一行，如下所示:

import itertools
s = StringIO(''.join(itertools.dropwhile(lambda l: ':' in l, lines)))

(假设标题之后没有“坏”行)。