numpy寻找list或者数组某一维相同元素的索引

问题描述

在做检测任务时，想要寻找相同语义类别的所有框的索引。
描述框的数组为K*4，其中最后一个维度代表语义类别，目的是找出相同语义类别（即最后一维各种相同元素）的索引。

解决方法

• 主要使用一个for循环以及np.argwhere()函数来解决。

import numpy as np
list = [1, 2, 3, 4, 2, 2,