Numpy Array Square Each Element In An