結果 : take input for matrix in python