重新排列numpy 2D数组的列

Question

有没有办法将numpy 2D数组中的列顺序更改为新的任意顺序？例如，我有一个数组

array([[10, 20, 30, 40, 50],
       [ 6,  7,  8,  9, 10]])

我想将其更改为

array([[10, 30, 50, 40, 20],
       [ 6,  8, 10,  9,  7]])

通过应用排列

0 -> 0
1 -> 4
2 -> 1
3 -> 3
4 -> 2

在列上。因此，在新矩阵中，我希望将原始文档的第一列保留在原位置，将第二列移至最后一列，依此类推。

有一个numpy函数可以做到吗？我有一个相当大的矩阵，并且希望得到更大的矩阵，因此我需要一个可以在可能的情况下快速且适当地执行此操作的解决方案（置换矩阵是不可行的）

谢谢。

Answer 1

使用花式索引可以在 O（n） 时间和 O（n） 空间中实现：

>>> import numpy as np
>>> a = np.array([[10, 20, 30, 40, 50],
...               [ 6,  7,  8,  9, 10]])
>>> permutation = [0, 4, 1, 3, 2]
>>> idx = np.empty_like(permutation)
>>> idx[permutation] = np.arange(len(permutation))
>>> a[:, idx]  # return a rearranged copy
array([[10, 30, 50, 40, 20],
       [ 6,  8, 10,  9,  7]])
>>> a[:] = a[:, idx]  # in-place modification of a

请注意，a[:, idx]它返回的是副本，而不是视图。的
O（1） k-空间的解决方案是不可能的在一般情况下，由于如何numpy的阵列在内存中跨距。