結果 : write a matrix program in python