結果 : how to partition a matrix in python