python - 在 Python 中否定矩阵对角线的所有其他条目的最佳方法

Question

我想创建一个 numpy 矩阵，该矩阵与所有其他对角线元素取反。一些选项是:

丑陋的:

n = 10
i_star = np.eye(n)
for i in range(n):
    if i % 2 == 1:
        i_star[i,i] *= -1   

稍微好点:

n = 10
i_star = np.eye(n)
i_star = i_star[[np.arange(1,n,2)],[np.arange(1,n,2)]] *= -1

有没有更优雅的解决方案？

Answer 1

只需使用np.diag:

np.diag([1,-1]*5)