Python 按列的绝对值对多维数组排序

Python 如何按列的绝对值对numpy的多维数组排序？

在一些算法中需要通过对列的绝对值进行排序，但是如果直接使用sorted对多维数组排序的话会导致程序无法运行。因此经过查找多方资料和不断的实验，总结了这个问题。
首先，我们先得到一组数据：

我们要根据列的绝对值进行排序，首先我们要通过np.argsort去得到排序后的索引：

index = np.argsort(abs(mat),axis=0)
print(index)

然后根据 index 去得到排序后的数组：

new_mat = np.take_along_axis(mat, index, axis=0)
print(new_mat)

这样就达到了排序的目的。
Done ! ! !