[Python]Numpy配列(Array)を絶対値で降順にソートするには？

どうも、ちょげ(@chogetarou)です。

PythonのNumpy配列(Array)を絶対値で降順にソートする方法を紹介します。

方法

Numpy配列を絶対値で降順にソートするには、numpy.argsort()とnumpy.take_along_axis()を使います。を使います。

まず、np.argsort()を呼び出し、末尾に[::-1]を付けます。

np.argsort()の第1引数で引数に配列を指定したnumpy.abs()、引数「axis」にNoneを指定します。

index = np.argsort(np.abs(array), axis=None)[::-1]

次に、numpy.take_along_axis()を呼び出します。

take_along_axis()の第１引数にnumpy配列、第2引数に上記のnp.argsort()の結果を指定します。

引数「axis」にNoneを指定します。

そして、take_along_axis()からreshape()を呼び出し、引数にnumpy配列の形状を指定します。

result = np.take_along_axis(array, index, axis=None).reshape(array.shape)

上記のnumpy.take_along_axis().reshape()は、numpy配列を絶対値で降順にソートした結果を返します。

使用例

import numpy as np

nums = np.array([[8, -5, -6],  
                [-9, -2, 7], 
                [3, -1, -4],] )

index = np.argsort(np.abs(nums), axis=None)[::-1]

result = np.take_along_axis(nums, index, axis=None).reshape(nums.shape)

print(result)
#[[-9  8  7]
# [-6 -5 -4]
# [ 3 -2 -1]]