哪个 pandas DataFrame 行在应用 function 时发出警告？

Question

I have a DataFrame in which each row is one observation and each column a parameter. I need to apply a function to each row, and in some of them it raises a warning but I can't find out which ones. Example:

result = df.apply(lambda x: func(x, args), axis=1)

If df has 1000 rows, 500 will raise an identical warning message from func , but there's no way for me to know which ones.

Is there a way to output the DataFrame's rows that raised the warning and save them to a list, for example?

Answer 1

If this is for troubleshooting (vs production) you can use df.iterrows() to apply the function and print or log the row number, for example

for n, row in enumerate(df.iterrows()):
    print(row)
    func(row, args)