从几个 data.frame 列中创建一个因子变量

Question

我有一个data.frame，看起来像这样：

A <- data.frame(id = 1:5, col1 = c(0,1,1,0,0), col2 = c(2,0,0,0,0), col3 = c(0,0,0,3,3))

我如何用它创建一个factor 变量，所以它看起来像这样：

factor(c(2,1,1,3,3))

我知道如何从一列中提取一个因子，但不确定如何将它们合并在一起

Answer 1

您可以使用rowSums

A <- data.frame(id = 1:5, col1 = c(0,1,1,0,0), col2 = c(2,0,0,0,0), col3 = c(0,0,0,3,3))
A$col4 <- as.factor(rowSums(A[,2:4]))
str(A)

> str(A)
'data.frame':   5 obs. of  5 variables:
  $ id  : int  1 2 3 4 5
$ col1: num  0 1 1 0 0
$ col2: num  2 0 0 0 0
$ col3: num  0 0 0 3 3
$ col4: Factor w/ 3 levels "1","2","3": 2 1 1 3 3

Answer 2

您可以先将所有零转换为 NA，然后使用 coalesce 从 dplyr 将列“合并”为一个：

library(dplyr)

A$col4 = A %>%
  select(-id) %>%
  mutate_all(funs(replace(., . == 0, NA))) %>%
  {coalesce(!!! .)} %>%
  as.factor()

结果：

  id col1 col2 col3 col4
1  1    0    2    0    2
2  2    1    0    0    1
3  3    1    0    0    1
4  4    0    0    3    3
5  5    0    0    3    3

> A$col4
[1] 2 1 1 3 3
Levels: 1 2 3

注意： coalesce 中的!!! 表示法将参数拼接成点，所以它等价于coalesce(A$col1, A$col2, A$col3)