結果 : how to take a matrix input in python