python numpy数组元素之间的距离缩减

Question

我有一个像这样的python数组:

array([     18,      36,      54, ..., 9893804, 9893822, 9893840],
dtype=int64)

我想获得一个包含"距离"的数组。

在这个例子中应该是:(18岁,18岁,18岁,…， xxx, 18,18]

为了做到这一点，我使用了一个难看的for i in range(len(arr))

我怀疑有一种方法是使用np.flatnonzero(什么的)我找不到……有线索吗?

Answer 1

np.diff(arr)呢?

arr = np.array(...)
print(np.diff(arr)) # [18 18 18 ... 18 18 18]