python top k问题

numpy的：

import numpy as np
top_k=3
arr = np.array([1, 3, 2, 4, 5])
top_k_idx=arr.argsort()[::-1][0:top_k]
print(top_k_idx)

这个效率比较高：

1万条排序，不到1ms

import heapqimport numpy as npif __name__ == '__main__':a = [0, 1, 2, 3, 4, 5, 5, 7, 8, 10, 15, 20, 25]bbb= heapq.nsmallest(2, abs(np.array(a)-3.5))ccc= heapq.nsmallest(2,  enumerate(abs(np.array(a)-3.5)),key=lambda x: x[1])print(bbb)print(ccc)

#coding=utf-8
'''
Created on Sep 21, 2015
@author: haoyifen
'''
import time
import random
def left(i):return 2*i+1
def right(i):return 2*i+2
def min_heapify(a,heap_size,i):l=left(i)r=right(i)if l<heap_size and a[l]<a[i] :least=lelse:least=iif

python top k问题相关推荐

python top n排序
python top k 排序最小top n: import heapqimport numpy as npif __name__ == '__main__':a = [0, 1, 2, 3, 4, ...
从零开始用Python实现k近邻算法（附代码、数据集）
作者:Tavish Srivastava 翻译:王雨桐校对:丁楠雅本文约2000字,建议阅读8分钟. 本文将带领读者理解KNN算法在分类问题中的使用,并结合案例运用Python进行实战操作. 注意 ...
找出无序数组中最小的k个数（top k问题）
2019独角兽企业重金招聘Python工程师标准>>> 给定一个无序的整型数组arr,找到其中最小的k个数该题是互联网面试中十分高频的一道题,如果用普通的排序算法,排序之后自然可以 ...
海量数据处理的 Top K相关问题
全栈工程师开发手册 (作者:栾鹏) python数据挖掘系列教程 Top-k的最小堆解决方法问题描述:有N(N>>10000)个整数,求出其中的前K个最大的数.(称作Top k或者Top ...
Top K 问题的最优解 - 快速选择算法（图解详细教程）
文章目录 Leetcode 215. Kth Largest Element in an Array 1.1:快速选择算法流程 1.2:注意事项 1.3:python实现 Leetcode 973. ...
堆实战(动态数据流求top k大元素,动态数据流求中位数)
动态数据集合中求top k大元素第1大,第2大 ...第k大 k是这群体里最小的所以要建立个小顶堆只需要维护一个大小为k的小顶堆即可当来的元素(newCome)> 堆顶元素(smallTo ...
Top K算法问题的实现
前奏在上一篇文章,程序员面试题狂想曲:第三章.寻找最小的k个数中,后来为了论证类似快速排序中partition的方法在最坏情况下,能在O(N)的时间复杂度内找到最小的k个数,而前前后后upd ...
海量数据处理 - 10亿个数中找出最大的10000个数（top K）
海量数据处理 - 10亿个数中找出最大的10000个数(top K问题) 版权声明:本文为博主原创文章,未经博主允许不得转载前两天面试3面学长问我的这个问题(想说TEG的3个面试学长都是好和蔼,希望 ...
Python KNN K近邻分类
Python KNN K近邻分类 1 声明本文的数据来自网络,部分代码也有所参照,这里做了注释和延伸,旨在技术交流,如有冒犯之处请联系博主及时处理. 2 KNN简介相关概念见下: 对于给定的观测来 ...

python top k问题

python top k问题相关推荐

最新文章

热门文章