R将向量扩展为围绕中心值的相同长度

时间:2018-05-14 14:03:53

标签: r

假设我有一个清单:

[[1]]
 [1]  0.7125856 -1.4871811  0.6230076  1.0756424  0.8172592 -0.1327926 -0.7674947 -1.0738684  1.5706677
[10] -0.6674960

[[2]]
[1] -0.5778682 -1.1186447  0.8272505  0.5123162  0.6607654  1.6559877 -0.7961859 -0.8966686

[[3]]
 [1] -1.42867999 -0.21142100 -2.89762370  0.11036288  0.66846399 -0.77309213  0.35278541 -0.99579117 -0.73043142
[10] -0.01857864 -0.93654969  0.46751328

[[4]]
[1] -0.483896685 -0.207550362 -0.902920637 -0.008191134 -1.015636093

我还有一个带有所谓"中心"的索引的向量此列表中的向量值:

[1] 2 4 5 1

我想要获得的是给定范围(例如,3)获取围绕这些中心的向量子集,两侧的范围长度。因此,每个子集的长度必须为7,如果范围超出向量的长度,则必须添加NA:

[[1]]
[1]         NA         NA  0.7125856 [-1.4871811]  0.6230076  1.0756424  0.8172592

[[2]]
[1] -0.5778682 -1.1186447  0.8272505  [0.5123162]  0.6607654  1.6559877 -0.7961859

[[3]]
[1] -0.2114210 -2.8976237  0.1103629  [0.6684640] -0.7730921  0.3527854 -0.9957912

[[4]]
[1]           NA           NA           NA [-0.483896685] -0.207550362 -0.902920637 -0.008191134

我将中心放在括号中,因为SO不允许在代码中使用粗体文本。有没有办法在R中做这样的事情?

1 个答案:

答案 0 :(得分:1)

我认为您正在寻找类似的东西吗?

数据:

n <- 10
x <- list(rnorm(n),rnorm(n),rnorm(n),rnorm(n))
i <- c(2,4,5,1)
r <- 3

可能的解决方案:

f <- function(x) ifelse((x-r):(x+r) %in% 1:n,(x-r):(x+r),NA)
tmp <- lapply(i,f)
out <- mapply("[",x,tmp)
split(out,col(out))
$`1`
[1]         NA         NA  0.8120232  0.4980624 -2.9708413  0.6754918 -0.9900322

$`2`
[1] -1.0303741 -1.7432638  0.9466315 -1.3322712 -1.7281613 -0.1951453  0.3789493

$`3`
[1] -1.4120119  0.3675425 -0.8378275 -1.1080856 -0.9369994  1.0823376  1.2194782

$`4`
[1]         NA         NA         NA  0.1757164 -1.5768032  0.3121876  0.4443616