获取numpy数组中最大或最小的几个数

#!/usr/local/bin/python# -*- coding: UTF-8 -*-import numpy as npdef get_Several_MinMax_Array(np_arr, several):"""获取numpy数值中最大或最小的几个数:param np_arr:numpy数组:param several: 最大或最小的个数（负数代表求最大，正数代表求最小）:retur

cxg0508

6733人浏览 · 2020-07-09 14:58:07

cxg0508 · 2020-07-09 14:58:07 发布

#!/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))