Получить индексы соседних элементов, меньшие или равные порогу в Python

Question

У меня есть отсортированный массив расстояний, например:

d = np.linspace(0.5, 50, 200)

, и я хочу итеративно получить индексы соседних элементов, которые находятся на расстоянии, меньшем или равном L для каждого элемента d. Есть ли простой способ сделать это?

Answer 1

Вы можете использовать numpy .argwhere после проверки того, что последовательные различия меньше или равны L:

import numpy as np

np.argwhere((d[1:]-d[:-1])<=L)