7

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

    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
4

5 回答 5

20

这是你的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

更新

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

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

正确的或至少更合适的方法是使用转换输出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

谢谢大家的讨论！

于 2013-01-25T16:57:54.137 回答
6

我通常使用这种方法：

df[df > 0] = 1
于 2016-08-09T21:13:13.917 回答
4

使用可以使用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
于 2013-01-25T16:53:32.820 回答
1

只需使用一个比较：

d = t(matrix(c(3,0,2,0,1,1,1,2,2), 3))
d > 0
t(matrix(as.numeric(d>0), ncol(d)))
于 2013-01-25T16:48:23.027 回答
0 
> 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)})
于 2013-01-25T16:58:16.740 回答