Инверсия функции bincount numpy
Данный массив целых чиселc
Как я могу преобразовать это в массив целыхinds
такой, чтоnp.all(np.bincount(inds) == c)
правда?
Например:
>>> c = np.array([1,3,2,2])
>>> inverse_bincount(c) # <-- what I need
array([0,1,1,1,2,2,3,3])
Контекст: я пытаюсь отслеживать местоположение нескольких наборов данных, одновременно выполняя вычисления для всех из них. Я объединяю все данные вместе для пакетной обработки, но мне нужен индексный массив для извлечения результатов обратно.
Текущее решение:
def inverse_bincount(c):
return np.array(list(chain.from_iterable([i]*n for i,n in enumerate(c))))