Question

我试图通过编写插入聚合的函数来从for循环中毕业。它进展不顺利。

示例数据是：

group <- c(1, 1, 1, 2, 2, 2, 2, 2, 3, 1, 1, 1, 2, 2)
gdp <- c(3.5, 4.2, 5, 4, 4.2, 5, 5.5, 6, 3.5, 3.4, 4.0, 4.1, 4.3, 4.7)
df <- data.frame(group, gdp)

无法正常工作的函数目标是，对于组内x（gdp）的每个值，找到x与x的最小值之间的绝对差值，以及x与x的最大值之间的差值。 x，并返回较小的差异。如果组中只有1个值，或者该值是组中的第一个或最后一个项，则差值将为0，在这种情况下返回0。

订单很重要，因为我不希望它整理第1组的所有内容;我希望将函数应用于组中的每个gdp值，然后转到下一个组。

功能是：

get_dist <- function(x){  
    a <- abs(x - min(x)) 
    b <- abs(x -max(x))   
    c <-  ifelse(a < b, a, ifelse(a = 0), 0, b) 
    return(c)
}

然后是最后一步，使用聚合：

edge_dist <- with(df, aggregate(group, list(gdp), get_dist))

关于我哪里出错的任何建议？它没有回归我所期望的。

Answer 1

使用data.table而不是aggregate：

library(data.table)

# step 1: assign unique groups
u_grps <- rle(df$group)$lengths
df$id <- rep(1:length(u_grps), u_grps)

# step 2: calculate your row-level stuff using data.table
data.table(df)[, min_abs_diff:=pmin(abs(gdp-max(gdp)), abs(gdp-min(gdp))), 
               by=id][]

# result:
#     group gdp id  min_abs_diff
#  1:     1 3.5  1           0.0
#  2:     1 4.2  1           0.7
#  3:     1 5.0  1           0.0
#  4:     2 4.0  2           0.0
#  5:     2 4.2  2           0.2
#  6:     2 5.0  2           1.0
#  7:     2 5.5  2           0.5
#  8:     2 6.0  2           0.0
#  9:     3 3.5  3           0.0
# 10:     1 3.4  4           0.0
# 11:     1 4.0  4           0.1
# 12:     1 4.1  4           0.0
# 13:     2 4.3  5           0.0
# 14:     2 4.7  5           0.0

注意：此示例仅打印输出。如果要存储在对象中，请使用

之类的内容

df2 <- 
  data.table(df)[, min_abs_diff:=pmin(abs(gdp-max(gdp)), abs(gdp-min(gdp))), 
                 by=id]

聚合中用户定义函数的问题

1 个答案: