我想使用magrittr和dplyr的简洁性,根据其他列中的值在列子集中的行之间复制单个值。这是一个简单的示例;我想将这个想法应用于一个大型数据集的多个列,这些列在一长串命令中具有多个条件。
以数据帧df <- data.frame(a = 1:5, b = 6:10, x = 11:15, y = 16:20)
为例
a b x y
1 6 11 16
2 7 12 17
3 8 13 18
4 9 14 19
5 10 15 20
对于a = 5
所在的行,我想将x
和y
的值替换为b = 7
所在行中的值,如下所示:
a b x y
1 6 11 16
2 7 12 17
3 8 13 18
4 9 14 19
5 10 12 17
此尝试失败:
foo <- function(x){ifelse(df$a == 5, df[df$b == 7, .(df$x)], x)}
df %<>% mutate_each(funs(foo), x, y)
我能得到的最接近的是:
bar <- function(x){ifelse(df$a == 5, df[df$b == 7, "x"], x)}
df %<>% mutate_each(funs(bar), x, y)
但这是不正确的,因为它将这两个值分别替换为x
中的值,而不是x
和y
中的值。
谢谢你的建议。
发布于 2015-12-09 00:35:15
您可以使用mutate_each
和replace
来完成此操作
df %>% mutate_each(funs(replace(., a==5, nth(., which(b==7)))), x, y)
输出:
a b x y
1 1 6 11 16
2 2 7 12 17
3 3 8 13 18
4 4 9 14 19
5 5 10 12 17
或者根据@docendodiscimus的评论,它可以进一步缩短为(可能[
也比which
更好):
df %>% mutate_each(funs(replace(., a==5, .[b==7])), x, y)
发布于 2015-12-09 00:45:43
只要提到data.table
解决方案就可以了:
require(data.table)
setDT(df)[a == 5, c("x", "y") := df[b == 7, .SD, .SDcols = c("x", "y")]]
> df
a b x y
1: 1 6 11 16
2: 2 7 12 17
3: 3 8 13 18
4: 4 9 14 19
5: 5 10 12 17
或者,您也可以使用:
cols <- c("x", "y")
setDT(df)[a == 5, (cols) := df[b == 7, .SD, .SDcols = cols]]
# or
cols <- c("x", "y")
setDT(df)[a == 5, (cols) := df[b == 7, cols, with = FALSE]]
发布于 2015-12-09 00:50:16
如果您的主要需求是在较长的dplyr-管道中应用该函数,则可以执行类似以下示例的操作:
foo <- function(df, cols = c("x", "y")) {
df[df$a == 5, cols] <- df[df$b == 7, cols]
df
}
df %>% ... %>% foo(c("x", "y")) %>% ...
# a b x y
#1 1 6 11 16
#2 2 7 12 17
#3 3 8 13 18
#4 4 9 14 19
#5 5 10 12 17
https://stackoverflow.com/questions/34160769
复制相似问题