温馨提示:本文翻译自stackoverflow.com,查看原文请点击:r - Create new variable using a lookup table
lookup r sapply

r - 使用查找表创建新变量

发布于 2020-05-04 03:27:14

我想使用查找表创建一个新变量。数据框如下所示:

  id    sex     age length
   1    Female  1   45
   2    Female  2   54
   3    Female  3   56
   4    Female  4   60
   5    Female  5   60
   6    Female  6   61
   7    Female  7   63
   8    Male    1   55
   9    Male    2   54
   10   Male    3   58
   11   Male    4   61
   12   Male    5   65
   13   Male    6   63
   14   Male    7   65
   15   Male    8   67
   16   Male    9   68
   17   Male    10  69

查找表如下所示

sex    age  length
Female  1   50
Female  2   53
Female  3   56
Female  4   58
Female  5   60
Female  6   61
Female  7   63
Male    1   50
Male    2   54
Male    3   57
Male    4   60
Male    5   62
Male    6   63
Male    7   65
Male    8   66
Male    9   67
Male    10  69

我想创建一个growth.rate具有两个级别的新变量:“正常”和“低”,因此最终数据框如下所示:

id   sex   age  length  growth.rate
1   Female  1   45  Low
2   Female  2   54  Normal
3   Female  3   56  Low
4   Female  4   60  Normal
5   Female  5   60  Low
6   Female  6   61  Low
7   Female  7   63  Low
8   Male    1   55  Normal
9   Male    2   54  Low
10  Male    3   58  Normal
11  Male    4   61  Normal
12  Male    5   65  Normal
13  Male    6   63  Low
14  Male    7   65  Low
15  Male    8   67  Normal
16  Male    9   68  Normal
17  Male    10  69  Low

在此示例中,ID为1的growth.rate为“低”,因为其长度小于1岁女性的查找表中的值。

相反,id 2的growth.rate为“ Normal”,因为她的长度大于2岁女性的查找表中的值。

我想调整该解决方案但未成功 获得上下文堆栈溢出错误-for循环中嵌套的ifelse语句过多?

任何帮助深表感谢

查看更多

提问者
Chris
被浏览
22
akrun 2020-02-15 10:08

如果我们left_join基于“性别”,“年龄”在第一个和查找数据集之间进行比较,则会得到两个“长度”列,在这些列之间进行比较,并使用ifelse创建一个新列case_when

library(dplyr)
left_join(df1, lookup, by = c('sex', 'age')) %>%
    transmute(id, sex, age, 
      growth.rate = case_when(length.x <= length.y ~ "Low", 
        TRUE ~ "Normal"), length = length.x)
#   id    sex age growth.rate length
#1   1 Female   1         Low     45
#2   2 Female   2      Normal     54
#3   3 Female   3         Low     56
#4   4 Female   4      Normal     60
#5   5 Female   5         Low     60
#6   6 Female   6         Low     61
#7   7 Female   7         Low     63
#8   8   Male   1      Normal     55
#9   9   Male   2         Low     54
#10 10   Male   3      Normal     58
#11 11   Male   4      Normal     61
#12 12   Male   5      Normal     65
#13 13   Male   6         Low     63
#14 14   Male   7         Low     65
#15 15   Male   8      Normal     67
#16 16   Male   9      Normal     68
#17 17   Male  10         Low     69

在中data.table,可以使其更紧凑

library(data.table)
setDT(df1)[lookup, growth.rate := fcase(length <= i.length, "Low", 
           "Normal"), on = .(sex, age)]

或带有索引

setDT(df1)[lookup, growth.rate := 
       c("Normal", "Low")[1 + (length <= i.length)], on = .(sex, age)]

数据

df1 <- structure(list(id = 1:17, sex = c("Female", "Female", "Female", 
"Female", "Female", "Female", "Female", "Male", "Male", "Male", 
"Male", "Male", "Male", "Male", "Male", "Male", "Male"), age = c(1L, 
2L, 3L, 4L, 5L, 6L, 7L, 1L, 2L, 3L, 4L, 5L, 6L, 7L, 8L, 9L, 10L
), length = c(45L, 54L, 56L, 60L, 60L, 61L, 63L, 55L, 54L, 58L, 
61L, 65L, 63L, 65L, 67L, 68L, 69L)), class = "data.frame", row.names = c(NA, 
-17L))

lookup <- structure(list(sex = c("Female", "Female", "Female", "Female", 
"Female", "Female", "Female", "Male", "Male", "Male", "Male", 
"Male", "Male", "Male", "Male", "Male", "Male"), age = c(1L, 
2L, 3L, 4L, 5L, 6L, 7L, 1L, 2L, 3L, 4L, 5L, 6L, 7L, 8L, 9L, 10L
), length = c(50L, 53L, 56L, 58L, 60L, 61L, 63L, 50L, 54L, 57L, 
60L, 62L, 63L, 65L, 66L, 67L, 69L)), class = "data.frame", row.names = c(NA, 
-17L))