python矩阵每个元素n次方