Change values in a numpy array,a[a == 2] = 10