按最接近定义值的组进行子设置

Question

我有一个数据框，我想在每个组中 select y 最接近特定值的行（例如：5）。

set.seed(1234)
df <- data.frame(x = c(rep("A", 4),
                       rep("B", 4)),
                 y = c(rep(4, 2), rep(1, 2), rep(6, 2), rep(3, 2)),
                 z = rnorm(8))

df

##   x y          z
## 1 A 4 -1.2070657
## 2 A 4  0.2774292
## 3 A 1  1.0844412
## 4 A 1 -2.3456977
## 5 B 6  0.4291247
## 6 B 6  0.5060559
## 7 B 3 -0.5747400
## 8 B 3 -0.5466319

结果将是：

##   x y          z
## 1 A 4 -1.2070657
## 2 A 4  0.2774292
## 3 B 6  0.4291247
## 4 B 6  0.5060559

谢谢你，菲利普

Answer 1

df %>%
  group_by(x) %>%
  mutate(
    delta = abs(y - 5)
  ) %>%
  filter(delta == min(delta)) %>%
  select(-delta)

Answer 2

val <- 5
delta <- abs(val - df$y)
df <- df[delta == min(delta), ]

Answer 3

或者使用基数 R:

 df[do.call(c, tapply(df$y, df$x, function(x) x-5 == max(x - 5))),]
  x y          z
1 A 4 -1.2070657
2 A 4  0.2774292
5 B 6  0.4291247
6 B 6  0.5060559

Answer 4

这是 data.table 的选项。将 'data.frame' 转换为 'data.table' (setDT(df))，按 'x' 分组，我们创建得到 'y' 与 5 的绝对差，检查元素是 min 从差异中获取行索引 (.I)，提取作为行索引的列 ("V1") 并对数据集进行子集化。

library(data.table)
setDT(df)[df[, {v1 <- abs(y-5)
               .I[v1==min(v1)]}, x]$V1]
#   x y          z
#1: A 4 -1.2070657
#2: A 4  0.2774292
#3: B 6  0.4291247
#4: B 6  0.5060559

按最接近定义值的组进行子设置

Sub-setting by group closest to defined value

r

subset

dplyr