我知道argmax()返回沿轴的最大值索引.
我还知道,在多次出现最大值的情况下,返回与第一次出现相对应的索引.
当您想要找到最大值及其索引时,argmax()可以正常工作.如何编写numpy.argmode()函数？
换句话说,如何在numpy数组中计算模式值并获取第一次出现的索引？
只是这样每个人都知道没有numpy.argmode,但这种功能的功能是我所寻求的.
我知道该模式会有多次出现.我们应该能够让它像argmax一样运行,如果我们有多次出现,它只返回第一次出现的值和索引.
我想要的一个例子是：
a = numpy.array([ 6, 3, 4, 1, 2, 2, 2])
numberIWant = numpy.argmode(a)
print(numberIWant)
# should print 4 (the index of the first occurrence of the mode)
我试过用：
stats.mode(a)[0][0]
numpy.argwhere(a==num)[0][0]
这确实有效,但我正在寻找一种更有效,更简洁的解决方案.
有任何想法吗？
解决方法:
如果你想留在NumPy,你可以使用np.unique的一些额外回报来得到你想要的：
>>> _, idx, cnt = np.unique(a, return_index=True, return_counts=True)
>>> idx[np.argmax(cnt)]
4
编辑
为了提供正在发生的事情的一些背景… np.unique总是返回一个排序的唯一值数组.可选的return_index提供另一个输出数组,其中索引在其中发生每个唯一值的第一次出现.并且可选的return_counts提供额外的输出,其中包含每个唯一值的出现次数.使用这些构建块,您需要做的就是在最高计数发生的位置返回索引数组的项.
标签：python,numpy,math 来源： https://codeday.me/bug/20190829/1760785.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。