python - 使用 np.where()[0]

Question

我的代码检测阈值以下的所有点，然后定位起点和终点。

below = np.where(self.data < self.threshold)[0]


startandend = np.diff(below)
startpoints = np.insert(startandend, 0, 2)
endpoints = np.insert(startandend, -1, 2)
startpoints = np.where(startpoints>1)[0]
endpoints = np.where(endpoints>1)[0]
startpoints = below[startpoints]
endpoints = below[endpoints]

我真的没有在 np.where() 函数之后使用 [0] 在这里

Answer 1

below = np.where(self.data < self.threshold)[0]

表示:

take the first element from the tuple of ndarrays returned by np.where() and assign it to below.