我想根据行和列位置重新排列数组，我该如何在 Python 中做到这一点？

Question

在 R 中我设法做到了，想法是在 Python 中做到，如果我在 Python 中应用 M[index]，则顺序与 R 中的结果不同。

R 中的代码：

> M = matrix(c("a",0,"z",
+               0,0,"b",
+               "c","y",0), nrow = 3, byrow = TRUE)
> M
     [,1] [,2] [,3]
[1,] "a"  "0"  "z" 
[2,] "0"  "0"  "b" 
[3,] "c"  "y"  "0" 
> 
> index = c(3,2,1)
> 
> M[index,index]
     [,1] [,2] [,3]
[1,] "0"  "y"  "c" 
[2,] "b"  "0"  "0" 
[3,] "z"  "0"  "a" 
> 

Python 代码：

M = np.array([["a",0,"z"],
                  [0,0,"b"],
                  ["c","y",0]])
index = [2,1,0]

print(M[index])
array([['c', 'y', '0'],
       ['0', '0', 'b'],
       ['a', '0', 'z']], dtype='<U1')

Answer 1

您可以使用np.flipud 和np.fliplr：

>>> M = np.array([["a",0,"z"],
                  [0,0,"b"],
                  ["c","y",0]])
>>> np.fliplr(np.flipud(M))
array([['0', 'y', 'c'],
       ['b', '0', '0'],
       ['z', '0', 'a']], dtype='<U1')

或者您可以使用np.rot90 旋转两次 90 度：

>>> np.rot90(M, 2)
array([['0', 'y', 'c'],
       ['b', '0', '0'],
       ['z', '0', 'a']], dtype='<U1')

如果你更喜欢通过索引来做，你可以这样做：

>>> index = [2,1,0]
>>> M[index, ::-1]
array([['0', 'y', 'c'],
       ['b', '0', '0'],
       ['z', '0', 'a']], dtype='<U1')
# or,
>>> M[::-1, index]
array([['0', 'y', 'c'],
       ['b', '0', '0'],
       ['z', '0', 'a']], dtype='<U1')

本质上类似于：

>>> M[::-1, ::-1]
array([['0', 'y', 'c'],
       ['b', '0', '0'],
       ['z', '0', 'a']], dtype='<U1')