从特定行子集中改变 R 中的列

发布于05月15日

我正在try 基于其他三列在数据框中创建一个新列:父列、特定指示符列和组合的父指示符的值.

已给予:

  parent specific val
1      a        x  10
2      a        y  11
3      a        z  12
4      b        x  20
5      b        y  21
6      b        z  22
7      c        x  30
8      c        y  31
9      c        z  32

我希望创建一个新列，比如px_val( Select 每个父级的x值)，这样得到的数据帧是:

  parent specific val px_val
1      a        x  10     10
2      a        y  11     10
3      a        z  12     10
4      b        x  20     20
5      b        y  21     20
6      b        z  22     20
7      c        x  30     30
8      c        y  31     30
9      c        z  32     30

测试df代码:

df <- data.frame(
  parent=c('a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'c'),
  specific=c('x', 'y', 'z', 'x', 'y', 'z', 'x', 'y', 'z'),
  val=c(10, 11, 12, 20, 21, 22, 30, 31, 32)
)

我曾考虑迭代数据帧，将给定父对象的x值存储在一个变量中，并将其赋给每个父对象.但感觉必须有一个更优雅的解决方案？

推荐答案

我们可以这样做:

px_val将包含值，其中specific等于每个唯一父级的x->；val[specific == 'x']

仅针对此变异的.by=...个组，其优点是此后不需要ungroup():

library(dplyr) #>= dplyr 1.1.0

df %>%
  mutate(px_val = val[specific == 'x'], .by=parent)

  parent specific val px_val
1      a        x  10     10
2      a        y  11     10
3      a        z  12     10
4      b        x  20     20
5      b        y  21     20
6      b        z  22     20
7      c        x  30     30
8      c        y  31     30
9      c        z  32     30