r - 创建一个对向量执行逻辑测试并返回向量的动态变量

Question

我正在尝试创建一个变量，它将检查向量 x 的前五个元素并返回一个由 1 和 0 组成的向量。如果变量等于 X[i] 则为 1，否则为 0。
类似于 =((X1:$X5=X1)*1) 在excel中的东西

X <- c("A","B","C","D","E","F","G","H")
#i tried this but i had errors
T <- length(X)
vec <- rep(0,T)
for(i in 1:T){
  j <- 5
if(T-i<=j){
  j <- T-i}
else{j <- 5}

if(X[i] == X[i:(i+j)]){
  vec[i] <- 1}
  else{
    vec[i] <- 0
}
}

Answer 1

也许你正在寻找

(X[1:5] == X[1]) * 1
#> [1] 1 0 0 0 0

例如:

X <- c("A", "B", "A", "D", "A")
(X[1:5] == X[1]) * 1
#> [1] 1 0 1 0 1

编辑
对于符合条件的向量列表，您可以执行以下操作:

X <- c("A", "A", "B", "C", "A", "B", "C", "E", "C", "A")

h <- 5

lapply(seq(length(X) - 5), function(i) (X[i:(i + h)] == X[i]) * 1)
#> [[1]]
#> [1] 1 1 0 0 1 0
#> 
#> [[2]]
#> [1] 1 0 0 1 0 0
#> 
#> [[3]]
#> [1] 1 0 0 1 0 0
#> 
#> [[4]]
#> [1] 1 0 0 1 0 1
#> 
#> [[5]]
#> [1] 1 0 0 0 0 1

由 reprex package 创建于 2020-08-20| (v0.3.0)