是否有一种方法来分类数据点与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

在iuc中，您可以按照以下方式使用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])