python生成对角线为2的矩阵