美文网首页每天一个芝士点
python返回数组的索引和位置的区别

python返回数组的索引和位置的区别

作者: 抄书侠 | 来源:发表于2020-05-25 08:55 被阅读0次

给定数组

import numpy as np
A = np.array([[1,2],[3,4]])

如果想要找到最大值可以使用

max_A = np.max(A)

进一步,如果想要得到最大值所在的位置,根据英语,感觉argmax就可以做到

A.argmax(max_A)
#>>> 3

返回值是3,这是由于在python存储数组时可以理解为存成一个列向量加上一个形状,所以直接使用index返回的索引就是展开成列向量数的位置。
我们希望它返回的值是[1,1]
这个时候就需要用到where了

np.where(A == np.max(A))

其中A == np.max(A)返回的是一个mask,结合where就可以解决问题了。

相关文章

  • python返回数组的索引和位置的区别

    给定数组 如果想要找到最大值可以使用 进一步,如果想要得到最大值所在的位置,根据英语,感觉argmax就可以做到 ...

  • Where do I belong(数组排序并找出元素索引)

    数组排序并找出元素索引 我身在何处? 先给数组排序,然后找到指定的值在数组的位置,最后返回位置对应的索引。 举例:...

  • 数组排序并找出元素索引

    数组排序并找出元素索引 我身在何处? 先给数组排序,然后找到指定的值在数组的位置,最后返回位置对应的索引。 举例:...

  • FCC-JS-BAS-Where do I belong-数组排

    数组排序并找出元素索引 我身在何处? 先给数组排序,然后找到指定的值在数组的位置,最后返回位置对应的索引。 举例:...

  • 数组的索引和切片

    索引: 获取数组中特定位置元素的过程切片: 获取数组元素子集的过程 一维数组的索引和切片: 与Python的列表类...

  • 7.Python中find方法和index方法的区别?

    Python中find方法和index方法的区别? find方法和index方法都是用来查找目标字符串的索引位置,...

  • Leetcode 35 搜索插入位置

    搜索插入位置 题目 给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回...

  • js中for in、for of、forEach

    遍历数组 for in返回数组的索引 for of返回数组的元素 遍历对象 for in返回对象的键 for of...

  • js字符串方法整理

    charAt() 返回指定索引位置的字符 charCodeAt() 返回指定索引位置的Unicode值 conca...

  • 搜索插入位置

    搜索插入位置 给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会...

网友评论

    本文标题:python返回数组的索引和位置的区别

    本文链接:https://www.haomeiwen.com/subject/ytyeahtx.html