开发者社区 问答 正文

是否有NumPy函数返回数组中某物的第一个索引?

我知道有一种方法可以让Python列表返回某些内容的第一个索引:

l = [1, 2, 3] l.index(2) 1

NumPy数组有类似的东西吗?

展开
收起
保持可爱mmm 2020-01-14 14:56:05 412 分享 版权
1 条回答
写回答
取消 提交回答
  • 是的,这是给定NumPy数组array和值的答案item,以搜索:

    itemindex = numpy.where(array==item)

    结果是具有所有行索引,然后是所有列索引的元组。

    例如,如果一个数组是二维的,并且它在两个位置包含您的商品,则

    array[itemindex[0][0]][itemindex[1][0]]

    将等于您的项目,因此

    array[itemindex[0][1]][itemindex[1][1]]

    问题来源于stack overflow

    2020-01-14 14:56:30
    赞同 展开评论