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

发布于 2021-02-02 23:18:13

我知道有一种用于Python列表的方法来返回某些内容的第一个索引:

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

NumPy数组有类似的东西吗?

关注者
0
被浏览
90
1 个回答
  • 面试哥
    面试哥 2021-02-02
    为面试而生,有面试问题,就找面试哥。

    是的,在给定NumPy数组array和值的情况下,这是搜索的答案item

    itemindex = numpy.where(array==item)
    

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

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

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

    将等于你的项目,因此

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


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看