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

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

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

方法

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

まず、np.argsort()を呼び出します。

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

index = np.argsort(np.abs(array), axis=None)

次に、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)

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

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