結果 : how to write code for matrix in python