How To Square An Array In Python