使用Python程序打印身份矩阵