如何对角线对矩阵进行分类

Question

如何转到：

0 2 3
1 5 1
5 6 8

to

0 1 3
1 2 5
5 6 8

或类似的东西？

Answer 1

怎么样：

a = np.random.randint(0,20,(n,n))
a
# array([[10,  4,  6, 12,  6],
#        [ 5, 19,  1, 11,  5],
#        [ 6, 10, 11,  1,  2],
#        [ 5,  3,  0,  9,  1],
#        [13, 13,  3, 15,  3]])
io = np.argsort(sum(np.ogrid[:n,:n]).ravel())
out = np.empty_like(a)
out.ravel()[io] = np.sort(a.ravel())
out
# array([[ 0,  1,  1,  3,  6],
#        [ 1,  2,  3,  6, 10],
#        [ 3,  5,  6,  9, 11],
#        [ 4,  5, 10, 12, 13],
#        [ 5, 11, 13, 15, 19]]) 

我们在X和Y索引的总和（使用ogrid生成）的总和上使用argsort来建立对角顺序。然后，使用高级索引将a的排序元素按以下顺序放置。