Replacing Values In Numpy Which Function