熊猫read_csv()有条件地跳过标题行

时间:2018-12-12 08:49:40

标签: python pandas csv

我正在尝试读取csv文件,但是我的csv文件有所不同。有些格式不同,有些则其他。我正在尝试添加控件,以便不需要编辑代码或输入文件。

我的问题是,其中某些csv文件在列标题上方有一行String。一个例子:

Created on 12-11-2018,CryptoDataDownload.com
Date,Symbol,Open,High,Low,Close,Volume From,Volume To
2018-12-11 11-AM,ADABTC,8.6e-06,8.61e-06,8.55e-06,8.57e-06,301141.7,2.59
2018-12-11 10-AM,ADABTC,8.69e-06,8.72e-06,8.6e-06,8.6e-06,236949.63,2.05

如果导入此文件,则分隔符将使用第一行并将文件分为Created on 12-11-2018CryptoDataDownload.com两列。

df.head()的外观如下:

                        Created on 12-11-2018 CryptoDataDownload.com
Date             Symbol Open     High     Low      Close              Volume From                          Volume To
2018-12-11 11-AM ADABTC 8.6e-06  8.61e-06 8.55e-06 8.57e-06              301141.7                               2.59
2018-12-11 10-AM ADABTC 8.69e-06 8.72e-06 8.6e-06  8.6e-06              236949.63                               2.05
2018-12-11 09-AM ADABTC 8.7e-06  8.7e-06  8.62e-06 8.69e-06             509311.39                               4.41
2018-12-11 08-AM ADABTC 8.69e-06 8.7e-06  8.63e-06 8.7e-06              111367.34                             0.9656

我要检查此文件是否具有此行,如果有,请跳过该行。

我该怎么做?

2 个答案:

答案 0 :(得分:3)

如果CSV文件中的标题遵循类似的模式,则可以执行一些简单的操作,例如在确定是否跳过第一行之前先嗅出第一行。

filename = '/path/to/file.csv'
skiprows = int('Created in' in next(open(filename)))
df = pd.read_csv(filename, skiprows=skiprows)

好的做法是使用上下文管理器,因此您也可以这样做:

filename = '/path/to/file.csv'
skiprows = 0
with open(filename, 'r+') as f:
    for line in f:
        if line.startswith('Created '):
            skiprows = 1
        break
df = pd.read_csv(filename, skiprows=skiprows)

答案 1 :(得分:0)

在pandas read_csv命令中使用'comment'参数时,可以跳过以特定字符开头的行。您可以使用以下代码跳过以“ C”开头的行:

filename = '/path/to/file.csv'
pd.read_csv(filename, comment = "C")
相关问题