为了日后方便,留个笔记贴,也可以为有需要的小白朋友做参考(本人刚接触Python,大佬轻喷)
2023/5/22
今日完成了准则5 topk的主要部分,今天python掌握的几个小知识点为:
从多维的ndarray中获取最大的前K个值的索引以及值:
import numpy as np
k = 5
arr = np.random.randint(0,100,size=(10,10,10,10))
flattened_arr = arr.flatten()
sorted_indices = np.argsort(flattened_arr)
topk_indices = sorted_indices[-k:]
topk_values = flattened_arr[topk_indices]
print("Top k values:", topk_values)