Numpy Index With Array Initialization