import numpy as np n = int(input('输入每个维度的元素个数：')) a = np.arange(1,n**2+1).reshape(n,n) res = np.trace(a) print('矩阵的迹为', res, sep='')怎么用for循环来达到同样的效果

可以使用for循环来遍历矩阵的对角线元素，然后将它们累加起来，最终得到矩阵的迹。修改后的代码如下：

import numpy as np 

n = int(input('输入每个维度的元素个数：'))
a = np.arange(1,n**2+1).reshape(n,n)

res = 0
for i in range(n):
    res += a[i][i]

print('矩阵的迹为', res, sep='')

这样输出的结果与之前使用np.trace()函数得到的结果应该是一致的。