Een NumPy-rij bevat groepsnummers voor een reeks studenten, die via de index van de rij geïdentificeerd worden. De groepsnummers zijn gehele getallen, en het laagste groepsnummer is 0. Schrijf een functie groepen() met als argument deze rij. Het resultaat is een lijst van NumPy-rijen. Deze lijst bevat evenveel elementen als er groepen zijn. Elk element van zo'n rij bevat de studentennummers die tot een welbepaalde groep behoren. Met andere woorden: rij 0 bevat alle studenten van groep 0, rij 1 alle studenten van groep 1, enz. De rijen zijn opklimmend gesorteerd. Je mag veronderstellen dat er geen groepsnummers worden overgeslagen.

Argumenten

Een rij gehele getallen, met minstens 1 element.

Resultaat

Een lijst van NumPy-rijen, waarbij elke rij studentenindices van dezelfde groep bevat.

Voorbeeld

groepen(np.array([0, 1, 0, 2, 1, 0, 2])) = [[0 2 5], [1 4], [3 6]]