python数组全部平方