関数の機能：指定した要素の前後に小さい値または大きい値が集まる様に並び替える

例1 : 要素を1つだけ指定

>>> a = np.array([3, 4, 2, 1, 8, 3, 3])
>>> a.partition(3)
>>> a
array([3, 1, 2, 3, 8, 4, 3])

この例では7要素の配列の4つ目の要素(インデックスは3)で仕切って並べ替えています。4つ目の要素は3で、1〜3つ目の要素は3以下に、5〜7つ目の要素は3以上になっていることが分かります。

例2 : 要素を2つ指定

>>> a = np.array([3, 4, 2, 1, 8, 3, 4, 7, 1, 9])
>>> a.partition((2,5))
>>> a
array([1, 1, 2, 3, 3, 4, 8, 7, 4, 9])

この例では10要素の配列の3つ目の要素(インデックスは2)と6つ目の要素(インデックスは5)で仕切って並べ替えています。仕切った2箇所についてそれぞれ、仕切った要素より前方はその要素以下に、後方はその要素以上になっていることが分かります。

関連項目
関数ndarray.argpartitionの使い方の例