python shape 求矩阵的列数