Sắp xếp lại Matrix Elements với NumPy

Question

Tôi có một ma trận NumPy mà tôi đã đơn giản hóa để minh hoạ:

 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]] 

nơi số tại "nút giao thông" là quan trọng, nhưng thứ tự của chúng là không đúng. Tôi muốn sắp xếp lại các hàng và cột sao cho thứ tự mới là [a, d, b, e, c, f] nhưng giá trị này mà tôi gọi là "giao lộ" là giống nhau.

Dưới đây tôi đã bắt đầu chuyển đổi ma trận theo cách tôi muốn. Điền hàng 'e' liên quan đến việc xem xét các giao điểm trên cho (e, a) (= 4), sau đó (e, d) (= 7), sau đó (e, b) (= 5), (e, e) , (e, c) và (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, 

Mọi người có thể đề xuất cách sắp xếp lại ma trận theo cách này không?

Answer 1

chỉnh sửa: Tôi chỉ cần stumbled trên một giải pháp NumPy sử dụng advanced indexing:

#     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] 

Dưới đây là một giải pháp Python tinh khiết có thể được chuyển giao cho 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] 

Kết quả:

>>> 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]] 

Đây là phiên bản mà sửa đổi A tại chỗ:

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

Answer 2

NumPy cung cấp many phương pháp để thao tác mảng, bao gồm rolling elements along an xis, rolling all axes, swap axes. Bạn có thể sử dụng kết hợp các yếu tố này để có được thứ tự các yếu tố mong muốn