Python numpy recarray.partition用法及代碼示例

本文簡要介紹 python 語言中 numpy.recarray.partition 的用法。

用法:

recarray.partition(kth, axis=- 1, kind='introselect', order=None)

重新排列數組中的元素，使第 k 個位置的元素的值位於它在排序數組中的位置。所有小於第 k 個元素的元素都被移到這個元素之前，所有等於或大於第 k 個元素的元素都被移到它的後麵。兩個分區中元素的順序未定義。

參數：

kth： int 或整數序列

要分區的元素索引。第 k 個元素值將處於其最終排序位置，所有較小的元素將移到它之前，所有相等或更大的元素都將移到它後麵。分區中所有元素的順序未定義。如果提供了第 k 個序列，它將立即將由第 k 個索引的所有元素分區到它們的排序位置。

axis： 整數，可選

要排序的軸。默認為 -1，表示沿最後一個軸排序。

kind： {‘introselect’}，可選

選擇算法。默認為‘introselect’。

order： str 或 str 列表，可選

當a是定義了字段的數組時，該參數指定首先比較哪些字段，第二個等。單個字段可以指定為字符串，不需要指定所有字段，但仍會使用未指定的字段，在他們在 dtype 中出現的順序，以打破關係。

注意：

有關不同算法的說明，請參見np.partition。

例子：

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

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