How To Square Every Element In Numpy Array Initialization