将数值转换为二进制 (0/1)

Question

我有一个数据框，其中包含不同人的不同种类水果的数量。 像下面

    apple  banana  orange
Tim     3       0       2
Tom     0       1       1
Bob     1       2       2

我怎样才能把它变成一个二进制矩阵，即如果一个人至少有一个水果，不管他有多少，那么我记录 1，如果没有，记录 0。如下所示

    apple  banana  orange
Tim     1       0       1
Tom     0       1       1
Bob     1       1       1

Answer 1

这是你的data.frame ：

x <- structure(list(apple = c(3L, 0L, 1L), banana = 0:2, orange = c(2L, 
1L, 2L)), .Names = c("apple", "banana", "orange"), class = "data.frame", row.names = c("Tim", 
"Tom", "Bob"))

还有你的矩阵：

as.matrix((x > 0) + 0)
    apple banana orange
Tim     1      0      1
Tom     0      1      1
Bob     1      1      1

更新

我不知道睡前快速发帖会引起任何讨论，但讨论本身很有趣，所以我想在这里总结一下：

我的直觉是简单地接受一个事实，即在 R 中的TRUE和FALSE之下，是数字1和0 。 如果您尝试（一种不太好的方法）检查​​等效性，例如1 == TRUE或0 == FALSE ，您将得到TRUE 。 我的快捷方式（这将会需要更多的时间比正确的，或者至少在概念上更正确的方式）是只需添加0到我的TRUE S和FALSE S，因为我知道，R.将强制逻辑向量数字。

正确的，或者至少是更合适的方法是使用as.numeric转换输出（我认为这就是@JoshO'Brien 打算写的）。 但是……不幸的是，这会删除输入的维度属性，因此您需要将结果向量重新转换为矩阵，事实证明，这仍然比我在答案中所做的添加0快。

阅读了评论和批评后，我想我会再添加一个选项——使用apply循环遍历列并使用as.numeric方法。 这比手动重新创建矩阵要慢，但比在逻辑比较中添加0稍快。

x <- data.frame(replicate(1e4,sample(0:1e3)))
library(rbenchmark)
benchmark(X1 = {
            x1 <- as.matrix((x > 0) + 0)
          },
          X2 = {
            x2 <- apply(x, 2, function(y) as.numeric(y > 0))
          },
          X3 = {
            x3 <- as.numeric(as.matrix(x) > 0)
            x3 <- matrix(x3, nrow = 1001)
          },
          X4 = {
            x4 <- ifelse(x > 0, 1, 0)
          },
          columns = c("test", "replications", "elapsed", 
                      "relative", "user.self"))
#   test replications elapsed relative user.self
# 1   X1          100 116.618    1.985   110.711
# 2   X2          100 105.026    1.788    94.070
# 3   X3          100  58.750    1.000    46.007
# 4   X4          100 382.410    6.509   311.567

all.equal(x1, x2, check.attributes=FALSE)
# [1] TRUE
all.equal(x1, x3, check.attributes=FALSE)
# [1] TRUE
all.equal(x1, x4, check.attributes=FALSE)
# [1] TRUE

谢谢大家的讨论！

Answer 2

我通常使用这种方法：

df[df > 0] = 1

Answer 3

使用可以使用ifelse 。 它应该适用于矩阵和数据帧，但是，结果值将是矩阵

> df <- cbind(aaple = c(3, 0 , 1), banana = c(0, 1, 2), orange = c(2, 1, 2))
> df
     aaple banana orange
[1,]     3      0      2
[2,]     0      1      1
[3,]     1      2      2

> ifelse(df>0, 1, 0)
     aaple banana orange
[1,]     1      0      1
[2,]     0      1      1
[3,]     1      1      1

Answer 4

只需使用比较：

d = t(matrix(c(3,0,2,0,1,1,1,2,2), 3))
d > 0
t(matrix(as.numeric(d>0), ncol(d)))

Answer 5

> pippo
  person apple banana orange
1    Tim     1      0      2
2    Tom     0      1      1
3    Bob     1      2      2
> cols <- c("apple", "banana", "orange")
> lapply(cols, function(x) {pippo[,x] <<- as.numeric(pippo[,x] >= 1)})