Python如何根据阈值选择数据?
我在下面有一个数组:
a=np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 8, 7, 6, 5, 4, 3, 2, 1, 2, 3, 4, 5, 6, 7, 8, 9, 8, 7, 6, 5, 4, 3, 2, 1])
我想要的是在此向量上用阈值选择数据。
问题1:以threshold=8为例,大于等于8的元素保留,小于8为0。
输出向量应该是这样的:
a_output = [0, 0, 0, 0, 0, 0, 0, 8, 9, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 9, 8, 0, 0, 0, 0, 0, 0, 0,]
问题2:以threshold=8为例,大于等于8的元素保留,并且向两边扩展保留两个数据,小于8为0。
输出向量应该是这样的:
a_output = [0, 0, 0, 0, 0, 6, 7, 8, 9, 8, 7, 6, 0, 0, 0, 0, 0, 0, 0, 0, 0, 6, 7, 8, 9, 8, 7, 6, 0, 0, 0, 0, 0,]
(希望能获得解答,不知道我的表述是否清晰)