группы кластеров чисел в массиве

Question

У меня есть массив, как:

A = [1,3,8,9,3,7,2,1,3,9,6,8,3,8,8,1,2] 

И я хочу, чтобы подсчитать количество «въездных кластеров», которые >5. В этом случае результат должен быть 4, потому что:

[1, 3, (8,9), 3, (7), 2, 1, 3, (9,6,8), 3, (8,8), 1, 2] 

Учитывая L длину массива, я могу сделать:

A = [1,3,8,9,3,7,2,1,3,9,6,8,3,8,8,1,2] 
A = np.array(A) 

for k in range(0,L): 
    if A[k]>5: 
     print k, A[k] 

и это дает мне все записи больше, чем 5. Но как я мог группировать каждый кластер чисел?

Answer 1

Вы можете использовать функцию groupby от itertools.

from itertools import groupby 

A = [1,3,8,9,3,7,2,1,3,9,6,8,3,8,8,1,2] 
result = [tuple(g) for k, g in groupby(A, lambda x: x > 5) if k] 

print(result) 
# [(8, 9), (7,), (9, 6, 8), (8, 8)] 

print(len(result)) 
# 4