Alternativa rápida para numpy.median.reduceat

Em relação a esta resposta , existe uma maneira rápida de calcular medianas em uma matriz que possui grupos com um número desigual de elementos? Por exemplo: data = [1.00, 1.05, 1.30, 1.20, 1.06, 1.54, 1.33, 1.87, 1.67, ... ] index = [0, 0, 1, 1, 1, 1, 2, 3, 3, ... ] E então eu quero calcular a...