如何仅选择虚拟变量列？

Question

有没有办法只选择数据框中的那些列，其中列中的值一次为 0 或 1？

我有一个包含各种值的数据框，包括各种字符串和数字。

我尝试使用 dplyr select，但找不到评估列中包含的值的方法。

示例数据如下所示。

data %>% 
  tribble(
    ~id, ~gender, ~height, smoking,
    1, 1, 170, 0,
    2, 0, 150, 0,
    3, 1, 160, 1
  )

Answer 1

您可以将函数（或 rlang-tilde 函数）传递给 select_if，并查找仅包含 0:1 的列。

tribble(
    ~id, ~gender, ~height, ~smoking,
    1, 1, 170, 0,
    2, 0, 150, 0,
    3, 1, 160, 1
  ) %>%
  select_if(~ all(. %in% 0:1))
# # A tibble: 3 x 2
#   gender smoking
#    <dbl>   <dbl>
# 1      1       0
# 2      0       0
# 3      1       1

如果您在虚拟变量列中可能有 NA，您可能希望在谓词中使用 %in% c(0:1, NA)。