如何在 R 中填充一行条件为另一行的值

  • 本文关键字:一行 条件 填充 r dplyr tidyr
  • 更新时间 :
  • 英文 :


我继承了一个以不寻常的方式编码的数据集。我想学习一种不那么冗长的重塑方式。数据框如下所示:

# Input.
participant  = c(rep("John",6), rep("Mary",6))
day          = c(rep(1,3), rep(2,3), rep(1,3), rep(2,3))
likes        = c("apples", "apples", "18", "apples", "apples", "7", "bananas", "bananas", "24", "bananas", "bananas", "3")
question     = rep(c(1,1,0),4)
number       = c(rep(18,3), rep(7,3), rep(24,3), rep(3,3))
df           = data.frame(participant, day, question, likes)
   participant day question   likes
1         John   1        1  apples
2         John   1        1  apples
3         John   1        0      18
4         John   2        1  apples
5         John   2        1  apples
6         John   2        0       7
7         Mary   1        1 bananas
8         Mary   1        1 bananas
9         Mary   1        0      24
10        Mary   2        1 bananas
11        Mary   2        1 bananas
12        Mary   2        0       3

如您所见,喜欢的列是异构的。当问题等于 0 时,喜欢传达的是参与者选择的数字,而不是他们喜欢的水果。所以我想在新列中重新编码它,如下所示:

   participant day question   likes number
1         John   1        1  apples     18
2         John   1        1  apples     18
3         John   1        0      18     18
4         John   2        1  apples      7
5         John   2        1  apples      7
6         John   2        0       7      7
7         Mary   1        1 bananas     24
8         Mary   1        1 bananas     24
9         Mary   1        0      24     24
10        Mary   2        1 bananas      3
11        Mary   2        1 bananas      3
12        Mary   2        0       3      3

我目前使用 base R 的解决方案涉及子集化初始数据框、创建查找表、更改列名称,然后将查找表与原始数据框合并。但这涉及几个步骤,我担心应该有一个更简单的解决方案。我认为tidyr可能是答案,但我不知道如何使用它将值分散在一列(喜欢)条件其他列(日期问题)中。

你有什么建议吗?多谢!

使用上面的数据集,您可以尝试以下操作。您可以按participantday对数据进行分组,并为每个组查找包含question == 0的行。

library(dplyr)
group_by(df, participant, day) %>%
mutate(age = as.numeric(as.character(likes[which(question == 0)])))

或者正如 alistaire 建议的那样,您也可以使用grep()

group_by(df, participant, day) %>%
mutate(age = as.numeric(grep('\d+', likes, value = TRUE)))

#   participant   day question   likes   age
#        (fctr) (dbl)    (dbl)  (fctr) (dbl)
#1         John     1        1  apples    18
#2         John     1        1  apples    18
#3         John     1        0      18    18
#4         John     2        1  apples     7
#5         John     2        1  apples     7
#6         John     2        0       7     7
#7         Mary     1        1 bananas    24
#8         Mary     1        1 bananas    24
#9         Mary     1        0      24    24
#10        Mary     2        1 bananas     3
#11        Mary     2        1 bananas     3
#12        Mary     2        0       3     3

如果你想使用data.table,你可以做:

library(data.table)
setDT(df)[, age := as.numeric(as.character(likes[which(question == 0)])),
            by = list(participant, day)]

注意

本数据集是一个新的数据集。Jota的答案适用于已删除的数据集。

处理新的示例数据:

# create a key column, overwrite it later
df$number <- paste0(df$participant, df$day) # use as a key
# create lookup table
lookup <- df[!is.na(as.numeric(as.character(df$likes))), c("number", "likes")]
# use lookup to overwrite df$number with the appropriate number
df$number <- lookup$likes[match(df$number, lookup$number)]
#   participant day question   likes number
#1         John   1        1  apples     18
#2         John   1        1  apples     18
#3         John   1        0      18     18
#4         John   2        1  apples      7
#5         John   2        1  apples      7
#6         John   2        0       7      7
#7         Mary   1        1 bananas     24
#8         Mary   1        1 bananas     24
#9         Mary   1        0      24     24
#10        Mary   2        1 bananas      3
#11        Mary   2        1 bananas      3
#12        Mary   2        0       3      3

由于将字符转换为数字 ( as.numeric(as.character(df$likes)) ),预计会通过强制引入 NA 的警告。


如果数据按示例中所示排序,则可以使用 zoo 包中的na.locf

library(zoo)
df$age <- na.locf(as.numeric(as.character(df$likes)), fromLast = TRUE)

最新更新