代码之家 › 专栏 › 技术社区 › Omry Atia

提取向量中与索引最近的相邻索引

vectorization r

Omry Atia · 技术社区 · 6 年前

N ,和一个随机索引 ind 在这个向量中。我想以最对称的方式提取11个最接近 ind公司 在最有效的方法中,比如不超过原始向量的长度,也不会得到负指数。

如果 N = 200 ,和 ind = 199

ind = 3 ,结果应为1:11。

如果 ind = 195 ,结果应为190:200。

我可以和你一起做 if

ff = function(ind, N){
  vv = sapply(ind, function(x) c((x-5):(x+5)))
  if (max(vv) > N) {
    vv = vv - max(vv) + N
  } else if (min(vv) < 6) {
    vv = vv - min(vv) + 1
  }
  return(vv)
}

有没有更有效的方法(例如子集)来实现这一点?

1 回复 | 直到 6 年前

chinsoon12 6 年前

以下是几种方法:

1) 使用 order

sort(head(order(abs(1:N - ind)), 11L))

2) 排序索引和 ind 然后进入前11个位置

sort((1:N)[as.integer(names(head(sort(abs(setNames(1:N, 1:N) - ind)), 11L)))])

3) 使用基数排序:

sort(head(sort(abs(1:N - ind), index.return=TRUE)$ix, 11L))

推荐文章

user3152377 · Python:矢量化/广播会提高速度吗?

6 年前

Remy · 如何在MATLAB中有效地求出阵列所选指标的最大值?

6 年前

Omry Atia · 在R中,对数据帧的行应用函数并返回数据帧

6 年前

Kapa11 · MATLAB:复制数组中的零项

6 年前

bumblebee · 回归“累积”期望值的矢量化

6 年前

Niklas M. · 如何在numpy中矢量化字符串数组的和?[副本]

6 年前

fales · Python中的许多对的相同操作

6 年前

user5054 · 有没有办法从矩阵中矢量化列的选择(重复)?

6 年前

Fnord · 具有屏蔽索引的numpy数组上的向量化和运算

6 年前

Rochak Agrawal · 以下八度音阶代码段的矢量化代码是什么

6 年前