Square Each Element Of Array Python Initialize Matrix Of Zeros