R：データから2値が入っている列名を得る

与えられたデータを２つのグループに分けるとき、２値の列を使ったりします。
ただ、列が多いときは、どの列が2値か得られると楽なので、その場合のメモです。

元データ

丁度良いサンプルデータが見つからなかったので、iris へテスト用に NA を含む2値の列を追加しました。

Sepal.Length.bool
TRUE または FALSE または NA
Petal.Width.ab
“A” または “B” または NA

library(dplyr)

data <- iris %>%
  mutate(
    Sepal.Length.bool = ifelse(Sepal.Length > 7, NA, trunc(Sepal.Length) %% 2 == 0),
    Petal.Width.ab = ifelse(Petal.Width > 2, NA, ifelse(trunc(Petal.Width) %% 2 == 0, "A", "B"))
  )

結果はこんな感じ。

data %>% tail(10)

2値の列を得る

cols <- colnames(data)
twoGroupCols <- cols[which(sapply(cols, function(col) {
  data[[col]] %>%
    na.omit() %>%
    unique() %>%
    length() == 2
}))]

で、２値の列が得られます。