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
"""
至此,我们得到了结果。
如果大家觉得有用,就请点个赞吧~