Ho una matrice numpy come questa [1,1,1,-1,-1,1,-1,1,1,-1,-1,-1,1,-1] Mi piacerebbe trovare la lunghezza della serie consecutiva più lunga di 1s o -1s. Nell'esempio, dovrebbe essere 3trova la lunghezza della serie di numeri consecutivi più lunga
in puro Python
>>> from itertools import groupby
>>> L = [1,1,1,-1,-1,1,-1,1,1,-1,-1,-1,1,-1]
>>> max(sum(1 for i in g) for k,g in groupby(L))
3
Non può essere più semplice di così. +1 –
Bel uso di itertools! – jszakmeister
@jszakmeister, ogni volta che il problema include la parola "consecutivo", 'groupby()' dovrebbe essere la prima cosa che fai su –
Simile alla risposta da @AlexMartelli
>>> import numpy as np
>>> nums = np.array([1,1,1,-1-1,1,-1,1,1,-1,-1,-1,1,-1])
>>> run_ends = np.where(np.diff(nums))[0] + 1
>>> np.diff(np.hstack((0, run_ends, nums.size))).max()
3
Vuoi un 'soluzione numpy' o una soluzione pura-python va bene ? È banale usare 'itertools.groupby' ... – Bakuriu
Quindi l'output in questo caso dovrebbe essere 2 (-1-1) giusto? ... ord c'è solo un", "mancante e in realtà vuoi 3 (1,1 1)? – pypat
@Bakuriu tutte le soluzioni pure-python sono soluzioni numpy. L'unica differenza è che a volte le soluzioni specifiche di numpy sono molto più belle o molto più veloci. –