python - 输出最大/最小的 k 个元素的索引

K = 4
a = np.array([0, 8, 0, 4, 5, 8, 8, 0, 4, 2])
# 最大的 k 个元素的索引
print(np.argpartition(a, -K)[-K:])   # [4 1 5 6]
# 最小的 k 个元素的索引
print(np.argpartition(a, K)[:K])     # [7 0 2 9]

python - 输出最大/最小的 k 个元素的索引相关推荐

输入 n 个整数，输出其中最小的 k 个
2019独角兽企业重金招聘Python工程师标准>>> /* 输入 n 个整数,输出其中最小的 k 个* TreeMap 和 TreeSet 是 Java Collection Fr ...
查找最小的k 个元素
排序一.题目:(感谢 http://blog.csdn.net/v_JULY_v 提供的题目) 查找最小的k 个元素输入n 个整数,输出其中最小的k 个. 例如输入1,2,3,4,5,6,7 和8 ...
在哪里能收到python实例代码-Python找出最小的K个数实例代码
题目描述输入n个整数,找出其中最小的K个数.例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,. 这个题目完成的思路有很多,很多排序算法都可以完成既定操作,关键是复 ...
程序员面试题精选100题(05)－查找最小的k个元素[算法]
题目:输入n个整数,输出其中最小的k个. 例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4. 分析:这道题最简单的思路莫过于把输入的n个整数排序,这样排在最前面的k个 ...
华为机试HJ58：输入n个整数，输出其中最小的k个
作者:翟天保Steven 版权声明:著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处题目描述: 输入n个整数,输出其中最小的k个. 本题有多组输入样例,请使用循环读入,比如whil ...
c语言最小元素下标怎么看,查找最小的k个元素（C语言代码）
解题思路: 1.对于初学者来说,可能会对n个整数用冒泡或者选择排序从小到大排序,然后输出最小的k个数即可. 2.但是如果数据量真的到1万或者10万的话,用普通排序肯定会超时哦,可以用桶排序来实现.如果 ...
python 无序列表中第k大元素_查询无序列表中第K小元素
当需要在无需列表中寻找第k小的元素时,一个显然的方法是将所有数据进行排序,然后检索k个元素.这种方法的运行时间为O(n log(n)). 无序列表调用分区函数将自身分解成两个子表,其长度为i和n-i. ...
python输出列表中的第二个元素_python – 替换列表中的每个第二个元素
我有一个二维列表: [[5, 80, 2, 57, 5, 97], [2, 78, 2, 56, 6, 62], [5, 34, 3, 54, 6, 5, 2, 58, 5, 61, 5, 16]] ...
python找出列表中所以相同元素的索引
在 Python 中可以使用 enumerate() 函数来找出列表中所有相同元素的索引.enumerate() 函数返回的是一个包含元素索引和元素值的元组,可以在循环中使用.示例如下: my_lis ...

python - 输出最大/最小的 k 个元素的索引

python - 输出最大/最小的 k 个元素的索引相关推荐

最新文章

热门文章