Finding range of a numpy array elements,"r = np.ptp(a, axis=1)"