如何从numpy中的数组中查找连续元素组？

Question

我必须将一个numpy数组中的连续元素聚类。考虑以下示例    a = [ 0, 47, 48, 49, 50, 97, 98, 99]输出应为元组列表，如下所示   [(0),(47, 48, 49, 50),(97, 98, 99)]这里的区别只是一个。之间的元素。如果差异也可以指定为限制或硬编码数字，那就太好了。非常感谢。

Answer 1

这是一个可能有用的功能：

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]]