Square Each Element Of Numpy Array Python