結果 : how to get identity matrix python