Inverso da função de contagem do numpy
Dada uma matriz de contagens de números inteirosc
, como posso transformar isso em uma matriz de números inteirosinds
de tal modo quenp.all(np.bincount(inds) == c)
é verdade?
Por exemplo:
>>> c = np.array([1,3,2,2])
>>> inverse_bincount(c) # <-- what I need
array([0,1,1,1,2,2,3,3])
Contexto: estou tentando acompanhar o local de vários conjuntos de dados, enquanto faço o cálculo de todos eles de uma só vez. Concateno todos os dados juntos para o processamento em lote, mas preciso de uma matriz de índice para extrair os resultados novamente.
Solução atual:
def inverse_bincount(c):
return np.array(list(chain.from_iterable([i]*n for i,n in enumerate(c))))