如何将LASSO筛选的二元指示变量行转换为标称标签变量行

你有一个1行30列的data.frame类型数据True_IVs，其中值为"0"或"1"，"1"代表选中的LASSO候选变量，需要提取这些变量的名称组成向量，之前代码报Error in s[[2]] : subscript out of bounds错误，以下是几种正确实现方式：

方法1：基础R直接筛选 #

利用逻辑索引直接提取符合条件的列名：

True_Regressors <- colnames(True_IVs)[True_IVs[1, ] == "1"]

原理：取True_IVs的第一行（数据为单行结构），生成值为"1"的逻辑向量，用这个向量筛选列名。

方法2：转为带名称的向量处理 #

先把单行数据转为带变量名的字符向量，再筛选：

# 提取第一行并转为字符向量
iv_vec <- as.character(True_IVs[1, ])
# 筛选值为"1"的元素名称
True_Regressors <- names(iv_vec)[iv_vec == "1"]

方法3：tidyverse风格实现 #

如果习惯用tidyverse工具链，可以这样写：

library(dplyr)
library(tibble)

True_Regressors <- True_IVs %>%
  t() %>%  # 转置，将列转为行
  as.data.frame() %>%
  rename(selected = V1) %>%  # 重命名值列
  filter(selected == "1") %>%  # 筛选选中的变量
  rownames_to_column("var_name") %>%  # 将行名转为列
  pull(var_name)  # 提取变量名字符向量

报错原因说明 #

你之前的代码出现下标越界错误，大概率是错误地将data.frame当成嵌套列表处理，试图访问不存在的列表元素（比如s[[2]]），但你的数据是单行多列的扁平data.frame结构，不存在二级列表元素，因此触发了该错误。

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