<分区>
我在这里试图完成的是生成一个大小为 n x n 的矩阵。无论矩阵是什么，我都必须从左上角到右下角填写数字 1，在其他任何地方填写 0。
def identity(m):
new_identity = []
old_identity = m
for i in range(len(old_identity)):
new_identity.append(old_list[1])
return new_identity
例如，如果矩阵为 3，则预期结果为:
[[1, 0, 0], [0, 1, 0], [0, 0, 1]]
或者为了更容易可视化:
[[1, 0, 0],
[0, 1, 0],
[0, 0, 1]]