Clamp Function Numpy Python