結果 : define identity matrix in python