识别二进制列

Question

我想识别data.frame中的二进制列。

例如，此表

my.table <-read.table(text="a,b,c
0,2,0
0.25,1,1
1,0,0", header=TRUE, as.is=TRUE,sep = ",")

会给FALSE, FALSE, TRUE

Answer 1

apply(my.table,2,function(x) { all(x %in% 0:1) })

（或

apply(my.table,2,function(x) { all(na.omit(x) %in% 0:1) })

如果您想允许NA值）

Answer 2

如果你想接受带有NA的二进制列，下面应该可以做到这一点：

is.binary <- function(v) {
  x <- unique(v)
  length(x) - sum(is.na(x)) == 2L
}

my.table <- data.frame(a=11:15, b=c(T,F,T,NA,T), c=c('foo',NA,'bar','bar','foo'))
vapply(my.table, is.binary, logical(1))
#    a     b     c 
#FALSE  TRUE  TRUE 

...或者如果你只接受0,1，NA：

is.binary <- function(v) {
  x <- unique(v)
  length(x) - sum(is.na(x)) == 2L && all(x[1:2] == 0:1)
}