pandas drop inf code example

Example 1: python pandas drop

df = pd.DataFrame(np.arange(12).reshape(3, 4),
...                   columns=['A', 'B', 'C', 'D'])
>>> df
   A  B   C   D
0  0  1   2   3
1  4  5   6   7
2  8  9  10  11

Drop columns
>>> df.drop(['B', 'C'], axis=1)
   A   D
0  0   3
1  4   7
2  8  11
>>> df.drop(columns=['B', 'C'])
   A   D
0  0   3
1  4   7
2  8  11

Example 2: df.drop

>>>df = pd.DataFrame(np.arange(12).reshape(3, 4), 
                     columns=['A', 'B', 'C', 'D'])
>>>df
   A  B   C   D
0  0  1   2   3
1  4  5   6   7
2  8  9  10  11

>>> df.drop(['B', 'C'], axis=1)
   A   D
0  0   3
1  4   7
2  8  11

OR

>>> df.drop(columns=['B', 'C'])
   A   D
0  0   3
1  4   7
2  8  11

Example 3: pandas dataframe replace inf

df.replace([np.inf, -np.inf], np.nan)

Example 4: check if dataframe contains infinity

df.replace([np.inf, -np.inf], np.nan).dropna(subset=["col1", "col2"], how="all")

Tags:

Python Example

Related