Быстрая альтернатива для numpy.median.reduceat

Что касается этого ответа , существует ли быстрый способ вычисления медиан по массиву, в котором есть группы с неодинаковым числом элементов? Например: 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, ... ] И затем я хочу вычислить разницу...