使用带有 read_csv 的转换器时获取错误行

Question

我有一个 csv 文件，其中包含 日期、Countries 等信息（只能采用预定义列表中的值），我目前正在使用 converters 参数进行一些健全性检查，如下所示：

import pandas as pd

def datesCheck(x):
    #do some checks and return the formated date
    pass

def countriesCheck(x):
    if x in countriesList:
        return x
    else:
    raise ValueError, x + ' is not a recognised country'

df = pd.read_csv('myCsvFile.csv', converters={'Dates': datesCheck, 'Countries': countriesCheck})

我想知道是否有办法获取引发（或第一个）ValueError（如果有）的行数，以便我可以打开 csv 文件并快速更改它。

感谢您的帮助。周末愉快！

Answer 1

引入一个计数器变量并计算行数并引发错误：

import pandas as pd

current_row = 0

def datesCheck(x):
    current_row += 1
    #do some checks and return the formated date
    # if date ok do nothing
    # else raise ValueError, 'Error Row = ' + str(current_row )
    pass

def countriesCheck(x):
    if x in countriesList:
        return x
    else:
    raise ValueError, x + ' is not a recognised country in row = ' + str(current_row)

df = pd.read_csv('myCsvFile.csv', converters={'Dates': datesCheck, 'Countries': countriesCheck})