Python Problema de iteração em vetores

[Ture]

Olá pessoal. Tudo bem!?

 

Estou enfrentando um problema simples mas que está me esquentando a cabeça, a questão é a seguinte:

Quero fazer uma função que receba dois vetores. um vetor de distâncias que chamarei de "dist" que pode ser iniciado aleatoriamente para testes. A função também recebe um outro vetor que é o de agrupamento, que chamarei de "grupo", ambos os vetores tem o mesmo tamanho e o vetor grupo obrigatoriamente tem valores inteiros contínuos(fora de ordem)começando em 0. A saída dessa função é a soma das distâncias conforme o agrupamento grupo

 

A saída desejada da função é a seguinte: EX:

dist = [5,5,10,3,2,7]

grupo = [1,1,2,2,0,0]

 

vetorsaida = [2+7, 5+5, 10+3]   ---> O índice do vetor grupo representa o valor do vetor distância que eu quero somar. Então todos os índices que possuem valor 0 no vetor devem ser somados(dist) e colocados na posição 0 da minha saída, todos os índices que possuírem valor 1 devem ser somados e colocados na posição 1 da minha saída.

 

Se puderem mandar as dicas e códigos em C, Python ou pseudo-código eu ficaria grato, mas caso alguém queira mandar a solução em outras linguagens também será bem vindo!

 

Outro ex de execução:

dist = [3,4,5,6,7,8,9]

grupo = [2,1,1,1,0,0,0]

 

vetorsaida = [7+8+9, 4+5+6, 3]

 

Espero que tenha ficado claro, agradeço pela atenção.

Ture!

adicionado 34 minutos depois

Problema resolvido, para os interessados segue o código da resolução:

 

dist = [1,2,3,4,5,6]
grupo = [0,0,1,1,2,2]
vetorsaida = [0,0,0]

for i in range(len(grupo)):
  for j in range(len(grupo)):
    if (i!=j and grupo[i] == grupo[j]):
      vetorsaida[grupo[i]]+= dist[j]