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

数组中的数字簇

1 个答案: