#!/usr/local/bin/python
# -*- coding: UTF-8 -*-
import numpy as np
def get_Several_MinMax_Array(np_arr, several):
"""
获取numpy数值中最大或最小的几个数
:param np_arr: numpy数组
:param several: 最大或最小的个数(负数代表求最大,正数代表求最小)
:return:
several_min_or_max: 结果数组
"""
if several > 0:
several_min_or_max = np_arr[np.argpartition(np_arr,several)[:several]]
else:
several_min_or_max = np_arr[np.argpartition(np_arr, several)[several:]]
return several_min_or_max
if __name__ == "__main__":
np_arr = np.array([0, 7, 0, 3, 5, 8, 8, 0, 3, 2])
print(get_Several_MinMax_Array(np_arr, -5))
运行结果:
[3 5 7 8 8]
本文介绍了一种使用Python和NumPy库获取数组中最大或最小几个数值的方法。通过argpartition函数,我们可以高效地找到指定数量的最大或最小元素,而无需完全排序整个数组,这在大数据集处理中尤其有效。
5417





