Square Each Element Of Array Python