странный цикл, почему не просто
```python
for i in range(n):
    d[i][i] = 0
```