R语言如何按条件从列表提取数据框指定行及列首尾元素

你现有代码存在3个核心问题：

• 命名冲突：你将存储多个数据框的列表对象命名为df，但df[[2]]本身是列表内的单个数据框，后续索引极易出错，建议先将列表重命名为df_list区分。
• 语法错误：你写的sapply代码中，函数入参定义为p_，内部却调用了不存在的p对象，同时末尾缺少闭合括号，运行会直接报错；且这段代码仅能返回行号，无法提取完整行内容。
• 逻辑错误：最初的unlist代码中which(p_$A)是筛选A列中为真值/非零值的位置，完全不符合提取A列首元素的需求，多余的右括号也会导致语法报错。

需求1：提取每个数据框D列首次为TRUE的行 #

用lapply遍历列表即可，额外加了无TRUE值的边界判断，避免返回异常：

# 提取所有符合条件的行，结果为和原列表等长的列表
first_true_rows <- lapply(df_list, function(dt) {
  match_idx <- which(dt$D == TRUE)[1]
  # 如果当前数据框D列没有TRUE值，返回空占位
  if (is.na(match_idx)) return(NULL)
  dt[match_idx, ]
})

# 如果需要将结果合并为一个汇总数据框，执行这行即可
first_true_total <- do.call(rbind, first_true_rows)

对你举的示例数据框，这段代码会准确返回D列第一个TRUE所在的第5行内容。

需求2：提取每个数据框A列的首尾元素 #

遍历列表时直接取A列的第一个、最后一个索引值即可，结果直接输出为结构化数据框：

A_head_tail <- as.data.frame(t(sapply(df_list, function(dt) {
  c(
    first_A = dt$A[1],
    last_A = dt$A[nrow(dt)]
  )
})))

输出结果的每一行对应原列表里一个数据框的A列首、尾值，不受各个数据框行数不同的影响。

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