結果 : how to take matrix input in python