извлечь argmin из 2d массива с условием - PullRequest
0 голосов
/ 24 сентября 2019

У меня есть двумерный массив, и я хочу получить индекс минимального значения, удовлетворяющего условию.Это мой код

import numpy as np

condition = np.where(costs_to_medoids < limit)
indice_min = np.unravel_index(np.argmin(costs_to_medoids [condition], axis = None), costs_to_medoids.shape)
...