python - 这是什么 : s[s[1:] == s[:-1]] do in numpy?

Question

我一直在寻找一种方法来有效地检查 numpy 数组中的重复项，并偶然发现了一个包含使用此代码的答案的问题。

这一行在 numpy 中是什么意思？

s[s[1:] == s[:-1]]

希望在应用之前了解代码。查看了 Numpy 文档，但找不到此信息。

Answer 1

切片 [1:] 和 [:-1] 表示除了第一个和除了最后一个 数组元素:

>>> import numpy as np
>>> s = np.array((1, 2, 2, 3))  # four element array
>>> s[1:]
array([2, 2, 3])  # last three elements
>>> s[:-1]
array([1, 2, 2])  # first three elements

因此，比较会在每个元素 s[x] 及其 "neighbour" s[x+1] 之间生成一个 bool 比较数组>，它将比原始数组短一个(因为最后一个元素没有邻居):

>>> s[1:] == s[:-1]
array([False,  True, False], dtype=bool)

并使用该数组来索引原始数组，您可以获得比较为 True 的元素，即与其邻居相同的元素:

>>> s[s[1:] == s[:-1]]
array([2])

请注意，这只标识相邻重复值。