shape计算矩阵的python代码示例