[python] How to change a single value in a NumPy array?