如何识别前后均为5的连续3整数序列并生成对应布尔向量？

嘿，这个问题用R的游程编码（rle）就能轻松搞定！我给你一步步拆解解决方案：

核心思路 #

我们需要找到被5前后包围的连续3序列，然后给这些序列对应的位置标记TRUE，其他位置标记FALSE。具体步骤：

1. 用rle函数把原序列拆成连续重复的元素块（比如c(5,3,3,5)会拆成5(1个)、3(2个)、5(1个)）
2. 逐个检查每个元素块：如果当前块是3，且它的前一个块是5、后一个块也是5，就把这个块的所有位置标记为TRUE
3. 最后生成和原序列长度一致的布尔向量

实现代码 #

# 示例输入向量
input_vec <- c(5,3,3,5,5,4,3,3,5)

# 生成游程编码，拆分连续重复元素
rle_obj <- rle(input_vec)

# 初始化结果布尔向量，默认全为FALSE
result <- logical(length(input_vec))

# 遍历每个游程块
for (i in seq_along(rle_obj$values)) {
  # 检查当前块是否是3，且前后块都是5（注意边界：第一个块没有前块，最后一个块没有后块）
  is_valid <- rle_obj$values[i] == 3 && 
              (i > 1 && rle_obj$values[i-1] == 5) && 
              (i < length(rle_obj$values) && rle_obj$values[i+1] == 5)
  
  if (is_valid) {
    # 计算当前块对应的起始和结束位置
    start_pos <- sum(rle_obj$lengths[1:(i-1)]) + 1
    end_pos <- start_pos + rle_obj$lengths[i] - 1
    # 标记这些位置为TRUE
    result[start_pos:end_pos] <- TRUE
  }
}

# 查看结果
result

代码解释 #

• rle(input_vec)：返回的对象包含两个部分，values是连续重复的元素值，lengths是每个元素连续出现的次数。比如示例的rle_obj$values是c(5,3,5,5,4,3,5)，lengths是c(1,2,2,1,2,1)
• 循环中判断is_valid：确保当前块是3，且不是第一个块（避免越界）、前一个块是5，同时不是最后一个块、后一个块是5
• 计算位置范围：通过累加前面块的长度，找到当前块在原序列中的起始和结束位置，批量标记为TRUE

测试示例 #

运行上面的代码，输出结果是：

[1] FALSE  TRUE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE

完全符合你给出的期望输出！

边界情况测试 #

比如输入c(5,3,5)，代码会输出c(F,T,F)，正确标记了中间被5包围的单个3；如果输入c(3,3,5)，因为前面没有5，输出全为FALSE，也符合要求。

内容的提问来源于stack exchange，提问作者otwtm