Comment trouver les groupes d'éléments consécutifs dans un tableau NumPy ?

Question

Je dois regrouper les éléments consécutifs d'un tableau NumPy. Considérons l'exemple suivant

a = [ 0, 47, 48, 49, 50, 97, 98, 99]

La sortie devrait être une liste de tuples comme suit

[(0), (47, 48, 49, 50), (97, 98, 99)]

Ici, la différence est juste une entre les éléments. Ce serait bien si la différence pouvait aussi être spécifiée comme une limite ou un nombre codé en dur.

Answer 1

def consecutive(data, stepsize=1):
    return np.split(data, np.where(np.diff(data) != stepsize)[0]+1)

a = np.array([0, 47, 48, 49, 50, 97, 98, 99])
consecutive(a)

donne

[array([0]), array([47, 48, 49, 50]), array([97, 98, 99])]

Answer 2

Voici une petite fonction qui pourrait vous aider :

def group_consecutives(vals, step=1):
    """Return list of consecutive lists of numbers from vals (number list)."""
    run = []
    result = [run]
    expect = None
    for v in vals:
        if (v == expect) or (expect is None):
            run.append(v)
        else:
            run = [v]
            result.append(run)
        expect = v + step
    return result

>>> group_consecutives(a)
[[0], [47, 48, 49, 50], [97, 98, 99]]
>>> group_consecutives(a, step=47)
[[0, 47], [48], [49], [50, 97], [98], [99]]

P.S. Ceci est du pur Python. Pour une solution NumPy, voir la réponse de unutbu.

Answer 3

(a[1:]-a[:-1])==1 produira un tableau booléen où False indique des ruptures dans les parcours. Vous pouvez également utiliser la fonction intégrée numpy.grad .

Answer 4

Voici ce que j'ai trouvé jusqu'à présent : je ne suis pas sûr que ce soit 100% correct.

import numpy as np
a = np.array([ 0, 47, 48, 49, 50, 97, 98, 99])
print np.split(a, np.cumsum( np.where(a[1:] - a[:-1] > 1) )+1)

retours :

>>>[array([0]), array([47, 48, 49, 50]), array([97, 98, 99])]

Answer 5

Testé pour les tableaux à une dimension

Obtenir où diff n'en est pas une

diffs = numpy.diff(array) != 1

Obtenez les index des différences, prenez la première dimension et ajoutez un à tous car diff par rapport à l'indice précédent

indexes = numpy.nonzero(diffs)[0] + 1

Fractionnement avec les index donnés

groups = numpy.split(array, indexes)