对于不同的 n 值,在 python 中创建 2D 列表(或 numpy 数组)的最佳方法是什么,其中对角线设置为 -1,其余值从 0 增加 1。例如,如果 n = 3,则数组将如下所示:
[[-1,0,1]
[2,-1,3]
[4,5,-1]]
或者 n = 4:
[[-1,0,1,2]
[3,-1,4,5]
[6,7,-1,8]
[9,10,11,-1]]
等等
我知道我可以创建一个包含零且对角线设置为 -1 的数组: a = numpy.zeros((n,n)) numpy.fill_diagonal(a,-1)
如果 n = 3 这将给出:
[[-1,0,0]
[0,-1,0]
[0,0,-1]]
但是我如何将 0 设置为递增的数字,如上面的示例所示?我需要通过循环迭代并设置值吗?或者有更好的方法来解决这个问题吗?
提前致谢。
最佳答案
一种方法 -
def set_matrix(n):
out = np.full((n,n),-1)
off_diag_mask = ~np.eye(n,dtype=bool)
out[off_diag_mask] = np.arange(n*n-n)
return out
示例运行 -
In [23]: set_matrix(3)
Out[23]:
array([[-1, 0, 1],
[ 2, -1, 3],
[ 4, 5, -1]])
In [24]: set_matrix(4)
Out[24]:
array([[-1, 0, 1, 2],
[ 3, -1, 4, 5],
[ 6, 7, -1, 8],
[ 9, 10, 11, -1]])
关于python - 在 numpy 数组中围绕定义的对角线设置递增值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47238708/