10

我有一个 NumPy 矩阵，我已经对其进行了简化以举例说明：

       a  b  c  d  e  f 
A =  [[0, 1, 2, 3, 4, 5],
 b    [1, 0, 3, 4, 5, 6],
 c    [2, 3, 0, 5, 6, 7],
 d    [3, 4, 5, 0, 7, 8],
 e    [4, 5, 6, 7, 0, 9],
 f    [5, 6, 7, 8, 9, 0]]

“十字路口”的数字很重要，但它们的顺序不正确。我想重新排列行和列，使新顺序为 [a, d, b, e, c, f] 但我称之为“交集”的这个值是相同的。

下面我已经开始按照我想要的方式转换矩阵。填充“e”行涉及查看上面的交叉点 (e,a) (= 4)，然后是 (e,d) (=7) ，然后是 (e,b) (=5)，(e,e) , (e,c) 和 (e,f)

       a  d  b  e  c  f
A1=  [[0, 3, 1, 4, 2, 5],
 d    [3, 0, 4, 7, 5, 8],
 b    [1, 4, 0, 5, 3, 6],  
 e    [4, 7, 5,

谁能建议如何以这种方式重新排列我的矩阵？

4

2 回答 2

17

编辑：我刚刚偶然发现了一个使用高级索引的 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]

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

#     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]
于 2012-06-07T17:48:54.287 回答
3

Numpy 提供了许多操作数组的方法，包括沿 xis 滚动元素、滚动所有轴、交换轴。您可以使用这些组合来获得所需的元素顺序

于 2012-06-07T17:54:23.983 回答