Возвращаем индекс того же объекта

Question

Я работаю с обнаружением объектов, потом обнаружил пример вывода многомерного массива ниже:

[[2433 1629 2656 1910]
 [2275  962 2538 1264]
 [2284 1191 2533 1472]
 [2303 1380 2551 1662]
 [2461 1442 2694 1732]
 [2284 1558 2533 1852]
 [2433 1261 2671 1524]
 [...................]
                      ]

Есть матрица [nx4], включающая в себя x, y, width, height. Теперь я хочу вернуть index/ data объекты, имеющие одинаковый идентификатор класса.Как я могу реализовать на Python и / или Numpy?

Answer 1

Представим, что вы сохранили классы в массиве classes с формой (n,), так что первая строка вашей матрицы M связана с первым элементом массива classes.

Для любого класса c вы можете найти все строки, которые попадают в класс c, используя

classes == c

Соответствующие строки в матрице задаются как

M[classes == c]