在没有NumPy的情况下访问python方阵中的对角元素

Question

我试图将矩阵主对角线上的所有元素都设置为等于1。这是我的相同代码，使用名为“ a”的矩阵：

for i in range(len(a)):
    a[i][i] = 1

但是，这将a中的所有元素都设置为1。我发现这是因为调用a [i] [i]正在访问整个第i行，而不是第i个对角元素。如何修改代码以达到预期的效果？