Exception Handling in Pandas .apply() function

后端 未结 3 1904
忘了有多久
忘了有多久 2021-02-18 16:15

If I have a DataFrame:

myDF = DataFrame(data=[[11,11],[22,\'2A\'],[33,33]], columns = [\'A\',\'B\'])

Gives the following dataframe (Starting ou

3条回答
  •  面向向阳花
    2021-02-18 16:32

    I had the same question, but for a more general case where it was hard to tell if the function would generate an exception (i.e. you couldn't explicitly check this condition with something as straightforward as isdigit).

    After thinking about it for a while, I came up with the solution of embedding the try/except syntax in a separate function. I'm posting a toy example in case it helps anyone.

    import pandas as pd
    import numpy as np
    
    x=pd.DataFrame(np.array([['a','a'], [1,2]]))
    
    def augment(x):
        try:
            return int(x)+1
        except:
            return 'error:' + str(x)
    
    x[0].apply(lambda x: augment(x))
    

提交回复
热议问题