python 二维矩阵转三维矩阵示例