如何基于特定值首次出现从矩阵中提取指定数量的连续行？

需求说明 #

从给定结构的矩阵中提取连续N行数据：

• 起始行是矩阵第11列（data[,11]）中首次出现指定日期的行
• 提取行数rows和起始日期start_date设为可配置变量

实现代码 #

首先定义配置变量：

start_date <- "2018-01-01_0130"
rows <- 3

加载你提供的示例矩阵：

data <- structure(c("278.499099686742", "281.836603879929", "281.253089979291", 
"279.783318564296", "285.362102091312", "283.65794852376", "283.811761811376", 
"285.933408588171", "284.710226729512", "290.374462082982", "274.311960190535", 
"281.336100324988", "246.696371361613", "252.707296982408", "254.526200145483", 
"242.377391904593", "252.634052559733", "246.359447017312", "251.322977393866", 
"255.605334639549", "255.405133217573", "252.419202253222", "236.290780380368", 
"252.707296982408", "256.567278057337", "263.586535230279", "264.538712725043", 
"251.811273545027", "263.386333808303", "256.567278057337", "261.543015837669", 
"266.616412848234", "266.010925620794", "263.066499829292", "244.728537872434", 
"263.828241825104", "275.249488800764", "280.872218981385", "279.824823737144", 
"271.653187647462", "285.252235457301", "279.585558623075", "281.902523860335", 
"284.365977942944", "283.047578334808", "286.931974217296", "263.159276098013", 
"281.602221727371", "258.989226967096", "262.522049620748", "260.720236822963", 
"256.784569844604", "266.367381811142", "262.187566757202", "263.295999020338", 
"264.2262031883", "262.856532484293", "267.361064478755", "250.966521203518", 
"262.522049620748", "275.911130085588", "283.792229965329", "281.411786228418", 
"271.560411378741", "288.211310133338", "280.593890175223", "284.453871250153", 
"286.477858796716", "285.012970343232", "289.41496014595", "261.003448590636", 
"283.626209273934", "274.294869825244", "283.277077525854", "281.353190690279", 
"269.106723219156", "286.971037909389", "278.870204761624", "283.513901159167", 
"286.17267370224", "284.934842959046", "287.598498463631", "257.824640646577", 
"283.140354603529", "264.392223879695", "272.93496504426", "271.467635110021", 
"258.933072909713", "274.861293360591", "268.108157590032", "272.446668893099", 
"274.622028246522", "273.779717385769", "274.622028246522", "248.781395927072", 
"273.054597601295", "236.3884396106", "241.122470796108", "243.874019607902", 
"232.84340955317", "241.283608525991", "236.075930073857", "240.187383666635", 
"244.611346796155", "244.41602833569", "241.015045642853", "228.392590135336", 
"241.068758219481", "276.218756660819", "283.076876103878", "280.50843834877", 
"272.676168084145", "287.669301405549", "281.116367056966", "283.831293657422", 
"285.41581466794", "283.633533716202", "289.070711359382", "263.435163423419", 
"282.781456932425", "2018-01-01_0115", "2018-01-01_0115", "2018-01-01_0115", 
"2018-01-01_0115", "2018-01-01_0115", "2018-01-01_0115", "2018-01-01_0115", 
"2018-01-01_0130", "2018-01-01_0130", "2018-01-01_0130", "2018-01-01_0130", 
"2018-01-01_0130", "0", "0", "0", "0", "0", "0", "0", "0", "0", 
"0", "0", "0"), dim = c(12L, 12L), dimnames = list(NULL, c("", 
"", "", "", "", "", "", "", "", "", "", "gt")))

执行提取逻辑：

# 定位第11列中首次出现目标日期的行索引
start_row <- which(data[,11] == start_date)[1]

# 边界检查：确保起始行有效且提取范围不超出矩阵
if (!is.na(start_row) && (start_row + rows - 1) <= nrow(data)) {
  # 提取连续rows行
  extracted_data <- data[start_row:(start_row + rows - 1), ]
  print(extracted_data)
} else {
  warning("未找到指定起始日期，或提取行数超出矩阵范围")
}

代码说明 #

1. which(data[,11] == start_date)[1]：筛选第11列匹配目标日期的所有行索引，取第一个值即为首次出现的位置
2. 边界检查避免报错：确认起始行存在，且连续提取的行不超过矩阵总行数
3. 通过行索引切片start_row:(start_row + rows - 1)完成连续行提取

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