Code to replace the inf and -inf values in a np.array for the max and min of the non inf values

replace_infs.py

import numpy as np

# To substitute inf value by the max non inf value

a = np.array([1, np.inf, 10, -5, -np.inf])
print(a)
# [  1.  inf  10.  -5. -inf] 

a[np.isposinf(a)] = a[~np.isposinf(a)].max()
a[np.isneginf(a)] = a[~np.isneginf(a)].min()

print(a)
# [  1.  10.  10.  -5. -5.]

def replace_infs(a):
    """Replaces the inf and -inf values in a np.array or pd.Series for the max 
    and min of the non inf values"""
    
    a[np.isposinf(a)] = a[~np.isposinf(a)].max()
    a[np.isneginf(a)] = a[~np.isneginf(a)].min()
    
    return a