How do assign a specific part of a Numpy array