Put Numpy Array Initialization Python Snake Code