結果 : how to define identity matrix in python