Python 获取数组中连续数据的组数

正文

今天遇到了这样一个问题，有以下数组：

arr = np.array([0, 1, 2, 5, 6, 7, 8, 9, 15, 16, 17])

我们想要获取它中间有多少个连续数据组成的组数，单个元素也算一组。代码如下：

import numpy as np

arr = np.array([0, 1, 2, 5, 6, 7, 8, 9, 15, 16, 17])

if len(arr) == 0:
    group = 0
else:  # len(arr) != 0:
    group = 1

for i in range(1, len(arr)):
    if arr[i] - arr[i - 1] > 1:
        group += 1

print(group)
"""
result:
3
"""

至此，我们得到了结果。

如果大家觉得有用，就请点个赞吧~