python-在定义的轴上使用最大值的索引

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python-在定义的轴上使用最大值的索引，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含626字，纯文字阅读大概需要1分钟。

内容图文

我想在矩阵“ A”的每一行的最大值的相同索引处提取数组“ B”的值.查找索引,我使用numpy函数“ numpy.argmax”,如下所示：

>>> A=numpy.asarray([[0,1,6],[3,2,4]]);A
array([[0, 1, 6],
       [3, 2, 4]])
>>> argA=numpy.argmax(A,axis=1);argA
array([2, 2])

问题是我不知道如何使用“ argA”提取数组“ B”中的值

解决方法:

argA中的每个条目对应于对应行中最大值的索引位置.这些行不是明确的(由于使用axis = 1),但对应于每个条目的索引.因此,您需要添加它们以获取所需的元素.

>>> A[[0,1], argA]
array([6, 4])

所以：

>>> B
array([[ 9,  8,  2],
       [ 3,  4,  5]])
>>> B[[0,1], argA] = 84,89
>>> B
array([[ 9,  8, 84],
       [ 3,  4, 89]])

概括使用：

>>> B[np.arange(A.shape[0]),argA]

内容总结

以上是互联网集市为您收集整理的python-在定义的轴上使用最大值的索引全部内容，希望文章能够帮你解决python-在定义的轴上使用最大值的索引所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端