python numpy数组减少元素之间的距离[重复]

Question

我在 python 中有一个这样的数组：

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

我希望获得一个数组，其中包含 ech 字节之间的“距离”...

在这种情况下，它将是： [18,18,18, ..., xxx, 18, 18]

为此，我使用了一个丑陋的for i in range(len(arr))

我怀疑有一种方法可以使用 np.flatnonzero(something) 我找不到.. 有什么线索吗？

Answer 1

np.diff(arr)怎么样？

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