如何对角排序矩阵

Question

如何前往：

0 2 3
1 5 1
5 6 8

到

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索引（使用argsort生成）的总和使用ogrid来建立对角线顺序。然后使用高级索引将a的已排序元素按此顺序放置。