How To Loop Over Numpy Array Initialization In Python