Numpyのwhereで他のインデックスの要素を条件にする方法

Question

前提・実現したいこと

Numpyにて，特定の要素を抽出したいとき，np.whereを利用するかと思うのですが，np.whereでは，その要素に対する条件は設定出来ますが，その要素の前の要素やインデックスを利用した条件は設定出来ないでしょうか．
例として

a = np.ndarray([1,2,3,4,5,7,11,18,22,25,32])

といった．配列から前の要素と比較して5以上大きい値のみを抽出したい場合は，どのような方法が考えられるでしょうか．

・for文
・np.roll
以外の方法が知りたいです．
np.rollでは計算量が増えるために利用したくありません．

Answer 1

一般的にうまく取り扱える高速な方法はなかなかないと思います。

今回のケースだと、こんな感じで書けるでしょうか。

>>> import numpy as np
>>> a = np.array([1,2,3,4,5,7,11,18,22,25,32])
>>> idx, = np.where(np.diff(a)>=5)
>>> idx += 1
>>> a[idx]
array([18, 32])

Comments

[taka_jp]

np.diffの存在は知りませんでした．
やはり一般的に添え字を使って書く方法はないのですね．
回答ありがとうございました．