python之Numpy 数组条件检查是否存在值为当前值 +1 的元素

我想将类似这样的 if 语句构建到 numpy 条件中(用于 argwhere 或类似条件)。

for i, element in enumerate(array): 
    if array[i]+1 == array[i+1]: 
        return True  

这是可能的还是循环遍历数组的唯一方法？

请您参考如下方法:

您可以使用移动数组的np.roll函数

import numpy as np 
array+1 == np.roll(array, shift=-1) 

这将返回一个 bool 值数组。