python - 有没有办法用 python 对高于值和系列阈值（连续数字高于阈值）的数据点进行分类？

Question

我试图通过首先建立一个阈值然后在一段时间内高于阈值时对它进行分类来使用 python 对数据进行分类。

示例数据：我的值阈值>4，我的系列阈值>2

[5,5,1,1,1,5,5,5,4,4,4,4,5,5,5,5,5] 

预期输出：注意前两个值为零，因为它们不满足系列阈值

[0,0,0,0,0,1,1,1,0,0,0,0,1,1,1,1,1]

所以，我只想在数据同时满足系列和值阈值时将其分类为 1。

有谁知道用python附加这个问题的最佳方法？

Answer 1

IIUC，您可以numpy按以下方式使用：

import numpy as np

a = np.array([5,5,1,1,1,5,5,5,4,4,4,4,5,5,5,5,5])
v_threshold = 4
s_threshold = 2

out = np.zeros_like(a)

out[s_threshold:] = a[s_threshold:] > v_threshold
#array([0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 0, 0, 1, 1, 1, 1, 1])