向量化删除非零元素

需要删除一个向量中非零值所在index小于5的元素,例如有如下的输入和输出向量
input  = 0 1 0 0 0 0 -1 0 1 0 0 0 0 1
output = 0 1 0 0 0 0  0 0 0 0 0 0 0 1
说明:input向量中的两个非零的值的索引8减去6小于5,所以它们两个符合条件置为0,而6和1等于5故保留。



请先 登录 后评论

1 个回答

peter
v.copy().fill!({x->x[0<=sub:P(x)<5] join x[sub:O(x)>-5]}(at(v != 0)), 0)
请先 登录 后评论
  • 1 关注
  • 0 收藏,141 浏览
  • wfHuang 提出于 2024-05-06 13:57

相似问题