python矩阵元素排序_python – 用Numpy重新排列矩阵元素

编辑：我偶然发现了一个使用

advanced indexing的NumPy解决方案：

# a b c d e f

A = numpy.array([[0, 1, 2, 3, 4, 5],

[1, 0, 3, 4, 5, 6],

[2, 3, 0, 5, 6, 7],

[3, 4, 5, 0, 7, 8],

[4, 5, 6, 7, 0, 9],

[5, 6, 7, 8, 9, 0]])

# a d b e c f

new_order = [0, 3, 1, 4, 2, 5]

A1 = A[:, new_order][new_order]

这是一个纯Python解决方案,可以转移到NumPy：

# a b c d e f

A = [[0, 1, 2, 3, 4, 5],

[1, 0, 3, 4, 5, 6],

[2, 3, 0, 5, 6, 7],

[3, 4, 5, 0, 7, 8],

[4, 5, 6, 7, 0, 9],

[5, 6, 7, 8, 9, 0]]

# a d b e c f

new_order = [0, 3, 1, 4, 2, 5] # maps previous index to new index

A1 = [[A[i][j] for j in new_order] for i in new_order]

结果：

>>> pprint.pprint(A1)

[[0, 3, 1, 4, 2, 5],

[3, 0, 4, 7, 5, 8],

[1, 4, 0, 5, 3, 6],

[4, 7, 5, 0, 6, 9],

[2, 5, 3, 6, 0, 7],

[5, 8, 6, 9, 7, 0]]

这是一个修改A的版本：

A[:] = [A[i] for i in new_order]

for row in A:

row[:] = [row[i] for i in new_order]