首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >R中带有magrittr和dplyr的条件数据帧突变

R中带有magrittr和dplyr的条件数据帧突变
EN

Stack Overflow用户
提问于 2015-12-09 00:16:37
回答 3查看 712关注 0票数 18

我想使用magrittr和dplyr的简洁性,根据其他列中的值在列子集中的行之间复制单个值。这是一个简单的示例;我想将这个想法应用于一个大型数据集的多个列,这些列在一长串命令中具有多个条件。

以数据帧df <- data.frame(a = 1:5, b = 6:10, x = 11:15, y = 16:20)为例

代码语言:javascript
复制
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所在的行,我想将xy的值替换为b = 7所在行中的值,如下所示:

代码语言:javascript
复制
a   b   x   y

1   6   11  16
2   7   12  17
3   8   13  18
4   9   14  19
5   10  12  17

此尝试失败:

代码语言:javascript
复制
foo <- function(x){ifelse(df$a == 5, df[df$b == 7, .(df$x)], x)}
df %<>%  mutate_each(funs(foo), x, y)

我能得到的最接近的是:

代码语言:javascript
复制
bar <- function(x){ifelse(df$a == 5, df[df$b == 7, "x"], x)}
df %<>%  mutate_each(funs(bar), x, y)

但这是不正确的,因为它将这两个值分别替换为x中的值,而不是xy中的值。

谢谢你的建议。

EN

回答 3

Stack Overflow用户

回答已采纳

发布于 2015-12-09 00:35:15

您可以使用mutate_eachreplace来完成此操作

代码语言:javascript
复制
df %>% mutate_each(funs(replace(., a==5, nth(., which(b==7)))), x, y)

输出:

代码语言:javascript
复制
  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更好):

代码语言:javascript
复制
df %>% mutate_each(funs(replace(., a==5, .[b==7])), x, y)
票数 13
EN

Stack Overflow用户

发布于 2015-12-09 00:45:43

只要提到data.table解决方案就可以了:

代码语言:javascript
复制
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

或者,您也可以使用:

代码语言:javascript
复制
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]]
票数 9
EN

Stack Overflow用户

发布于 2015-12-09 00:50:16

如果您的主要需求是在较长的dplyr-管道中应用该函数,则可以执行类似以下示例的操作:

代码语言:javascript
复制
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
票数 5
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/34160769

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档