結果 : how to write identity matrix in python