Condivisione della tecnologia

Applicazione Python Numpy topn

2024-07-08

한어Русский языкEnglishFrançaisIndonesianSanskrit日本語DeutschPortuguêsΕλληνικάespañolItalianoSuomalainenLatina

In Python, NumPy è una potente libreria di calcoli numerici.Quella che segue è un'implementazione della funzione TopN utilizzando NumPy

io

mport numpy as np

# 生成一个随机的整数数组
arr = np.random.randint(1, 100, 20)

# 定义获取 TopN 的函数
def top_n(arr, n):
    """
    该函数用于获取数组中的前 N 个最大值

    参数:
    arr (numpy.ndarray): 输入的数组
    n (int): 要获取的最大值的数量

    返回:
    numpy.ndarray: 包含前 N 个最大值的数组
    """
    # 对数组进行排序
    sorted_arr = np.sort(arr)[::-1]
    # 返回前 N 个值
    return sorted_arr[:n]

# 假设要获取前 5 个最大值
n = 5
print(top_n(arr, n))