重新排列和排序

Question

我有以下数据

ID v1 v2 v3 v4 v5 
1  1  3  6  4
2  4  2
3  3  1   8    5
4  2  5   3    1

我可以重新排列数据，使其自动创建新列并根据每个变量（v1 到 v5）中的值分配二进制值（1 或 0）吗？

例如在第一行中，我的值为 1、3、4 和 6。R 能否自动创建 6 个虚拟变量以将值分配给相应的列，如下所示：

 ID dummy1 dummy2 dummy3 dummy4 dummy5 dummy6
 1    1     0      1      1      0       1

要有这样的东西：

ID c1 c2 c3 c4 c5 c6 c7 c8
1  1  0  1  1  0  1  0  0
2  0  1  0  1  0  0  0  0
3  1  0  1  0  1  0  0  1
4  1  1  1  0  1  0  0  0

谢谢。

Answer 1

我们可以使用base R来做到这一点。遍历数据集除第一列以外的行，获取行中 max 值的序列，检查行中有多少个并将其转换为 integer 和 as.integer ，在末尾附加 NAs 以使 list 输出中的长度和 cbind 与第一列

中的长度相同

lst <- apply(df[-1], 1, function(x) as.integer(seq_len(max(x, na.rm = TRUE)) %in% x))
res <- cbind(df[1], do.call(rbind, lapply(lst, `length<-`, max(lengths(lst)))))
res[is.na(res)] <- 0
colnames(res)[-1] <- paste0('c', 1:8)
res
#  ID c1 c2 c3 c4 c5 c6 c7 c8
#1  1  1  0  1  1  0  1  0  0
#2  2  0  1  0  1  0  0  0  0
#3  3  1  0  1  0  1  0  0  1
#4  4  1  1  1  0  1  0  0  0

Answer 2

在基础 R 中，您可以使用：

table(transform(cbind(mydf[1], stack(mydf[-1]))[1:2], values = factor(values, 1:8)))
##    values
## ID  1 2 3 4 5 6 7 8
##   1 1 0 1 1 0 1 0 0
##   2 0 1 0 1 0 0 0 0
##   3 1 0 1 0 1 0 0 1
##   4 1 1 1 0 1 0 0 0

请注意，如果您希望“7”包含在输出中，则需要将堆叠值转换为 factor。这也适用于 "data.table" 和 "tidyverse" 方法。

或者，您可以使用 "data.table" 尝试以下操作：

library(data.table)
melt(as.data.table(mydf), "ID", na.rm = TRUE)[
  , dcast(.SD, ID ~ factor(value, 1:8), fun = length, drop = FALSE)]

或以下 "tidyverse":

library(tidyverse)
mydf %>% 
  gather(var, val, -ID, na.rm = TRUE) %>% 
  select(-var) %>% 
  mutate(var = 1, val = factor(val, 1:8)) %>% 
  spread(val, var, fill = 0, drop = FALSE)

示例数据：

mydf <- structure(list(ID = 1:4, v1 = c(1L, 4L, 3L, 2L), v2 = c(3L, 2L, 
    1L, 5L), v3 = c(6L, NA, 8L, 3L), v4 = c(4L, NA, 5L, 1L), v5 = c(NA, 
    NA, NA, NA)), .Names = c("ID", "v1", "v2", "v3", "v4", "v5"), row.names = c(NA, 
    4L), class = "data.frame")

如果自动化很重要，您还可以使用 "data.table" 方法中的 factor(value, sequence(max(value)) 或 "tidyverse" 方法中的 val = factor(val, sequence(max(val)))) 等语法。

Answer 3

另一个与 akrun 有一些相似之处的基本 R 答案是

# create matrix of values
myMat <- as.matrix(dat[-1])
# create result matrix of desired shape, filled with 0s
res <- matrix(0L, nrow(dat), ncol=max(myMat, na.rm=TRUE))
# use matrix indexing to fill in 1s
res[cbind(dat$ID, as.vector(myMat))] <- 1L
# convert to data.frame, add ID column, and provide variable names
setNames(data.frame(cbind(dat$ID, res)), c("ID", paste0("c", 1:8)))

哪个returns

  ID c1 c2 c3 c4 c5 c6 c7 c8
1  1  1  0  1  1  0  1  0  0
2  2  0  1  0  1  0  0  0  0
3  3  1  0  1  0  1  0  0  1
4  4  1  1  1  0  1  0  0  0

重新排列和排序

Rearrange and Sort

binary

r

reshape