Python Pandas – Check and Display row index with infinity

To check and display row index, use the isinf() with any(). At first, let us import the required libraries with their respective aliases −

import pandas as pd import numpy as np

Create a dictionary of list. We have set the infinity values using the Numpy np.inf −

d = { "Reg_Price": [7000.5057, np.inf, 5000, np.inf, 9000.75768, 6000, 900, np.inf] } 

Creating DataFrame from the above dictionary of list −

dataFrame = pd.DataFrame(d)

Getting row index with infinity values −

indexNum = dataFrame.index[np.isinf(dataFrame).any(1)] 

Example

Following is the code −

import pandas as pd import numpy as np # dictionary of list d = { "Reg_Price": [7000.5057, np.inf, 5000, np.inf, 9000.75768, 6000, 900, np.inf] } # creating dataframe from the above dictionary of list dataFrame = pd.DataFrame(d) print"DataFrame...\n",dataFrame # checking for infinite values and displaying the count count = np.isinf(dataFrame).values.sum() print"\nInfinity values count...\n ",count # getting row index with infinity values indexNum = dataFrame.index[np.isinf(dataFrame).any(1)] print"\nDisplay row indexes with infinite values...\n ",indexNum

Output

This will produce the following output −

DataFrame...     Reg_Price 0 7000.505700 1         inf 2 5000.000000 3         inf 4 9000.757680 5 6000.000000 6  900.000000 7       inf Infinity values count... 3 Display row indexes with infinite values... Int64Index([1, 3, 7], dtype='int64')