numpy获得前n大元素下标

http://blog.sina.com.cn/s/blog_4df4d7440102vu6u.html

参考：http://stackoverflow.com/questions/10337533/a-fast-way-to-find-the-largest-n-elements-in-an-numpy-array
http://stackoverflow.com/questions/6910641/how-to-get-indices-of-n-maximum-values-in-a-numpy-array

EDIT: Modified to include Ashwini Chaudhary’s improvement.

>>> import heapq
>>> import numpy
>>> a = numpy.array([1, 3, 2, 4, 5])
>>> heapq.nlargest(3, range(len(a)), a.take) [4, 3, 1]

For regular Python lists:

>>> a = [1, 3, 2, 4, 5] >>> heapq.nlargest(3, range(len(a)), a.__getitem__) [4, 3, 1]
If you use Python 2, use xrange instead of range.

或者：
找出矩阵a的前3个最大的元素的index:
a.argsort()[-3:][::-1]

Perhaps heapq.nlargest
import numpy as np import heapq x = np.array([1,-5,4,6,-3,3]) z = heapq.nlargest(3,x)
Result:
>>> z [6, 4, 3]
If you want to find the indices of the n largest elements using bottleneck you could use bottleneck.argpartsort
>>> x = np.array([1,-5,4,6,-3,3]) >>> z = bottleneck.argpartsort(-x, 3)[:3] >>> z array([3, 2, 5]

numpy获得前n大元素下标相关推荐

网易_在数组中查找前K个元素
笔试题,最后一题查找网易云音乐中播放量最大的前K个歌曲. 换句话说,就是在数组中查找前K大元素. 大致有以下几个思路. 1.第一感觉就是对数组进行降序全排序,然后返回前K个元素,即是需要的K个最大数 ...
数组中第K大元素（java多种方式实现）
题目描述: Leetcode215题:在未排序的数组中找到第 k 个最大的元素.请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素. 示例 1: 输入: [3,2,1,5 ...
【小f的刷题笔记】（JS）单调栈 - 下一个更大元素 LeetCode496 下一个更大元素的下标 LeetCode739 循环数组中下一个更大元素 LeetCode503
[单调栈] 单调栈:顾名思义,就是这个栈是单调的,后面也会运用这一点模版: var nextGreater = function (nums) {let n = nums.lengthlet res ...
数据结构：设线性表中每个元素有两个数据项k1和k2,现对线性表按一下规则进行排序：先看数据项k1，k1值小的元素在前，大的在后；在k1值相同的情况下，再看k2，k2值小的在前，大的在后。满足这种要求的
题目: 设线性表中每个元素有两个数据项k1和k2,现对线性表按一下规则进行排序:先看数据项k1,k1值小的元素在前,大的在后:在k1值相同的情况下,再看k2,k2值小的在前,大的在后.满足这种要求的排 ...
hdu1280 前m大的数（数组下标排序）
前m大的数 Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others) Total Subm ...
第十四周项目一数组大折腾（1）创建一个有20个元素的整型数组，通过初始化，为数组中的前10个元素赋初值，然后通过键盘输入后10个元素的值，从前往后输出数组中元素的值，每五个元素
问题及代码创建一个有20个元素的整型数组,通过初始化,为数组中的前10个元素赋初值,然后通过键盘输入后10个元素的值,从前往后输出数组中元素的值,每五个元素换一行. /*烟台大学计算机学院 201 ...
数据流中的第k大元素的golang实现
设计一个找到数据流中第K大元素的类(class).注意是排序后的第K大元素,不是第K个不同的元素. 你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器,它包含数据流中 ...
leetcode 703. 数据流中的第K大元素最小堆解法 c语言
如题: 设计一个找到数据流中第K大元素的类(class).注意是排序后的第K大元素,不是第K个不同的元素. 你的 KthLargest 类需要一个同时接收整数 k 和整数数组nums 的构造器,它包含 ...
堆排序海量数据求前N大的值
最(大)小堆的性质: (1)是一颗完全二叉树,遵循完全二叉树的所有性质. (2)父节点的键值(大于)小于等于子节点的键值堆的存储一般都用数组来表示堆,i结点的父结点下标就为(i – 1) / 2. ...

numpy获得前n大元素下标

numpy获得前n大元素下标相关推荐

最新文章

热门文章