R中的头部/尾部断裂分类算法

Question

我正在尝试在R (see here)中实现头/尾中断分类算法。与地图学中用于高度倾斜数据的其他分类方法相比，这种相对较新的算法是一种计算成本较低的替代方法。

到目前为止，我一直在寻找Python (see here)中的代码作为模板，并取得了相对成功的结果。下面是我在R中的实现：

# fake data to classify
pareto_data <- c()
for (i in 1:100){
  pareto_data[i] <- (1.0/i)^1.16
}

# head/tail breaks algorithm
ht <- function(data){
  ln <- length(data)
  mn <- mean(data)
  res <- append(c(),mn) # this is where I was hopping to store my output
  head <- subset(data,data>=mn)
  while (length(head)>=1 & length(head)/ln <= 0.40){
    print(res)
    return(ht(head))
  }
  #return(res)
}

ht(pareto_data)

由于运行上述代码，我已经能够打印出以下内容：

[1] 0.03849691
[1] 0.1779904
[1] 0.4818454 

此输出很可能与运行我用作模板的原始Python代码相同。但是，我还没有成功地将其存储在向量或列表中。

如果您能给出一些建议来克服这个问题并改进我的代码(这与Python语言中的原始代码不完全相同，特别是在while语句的条件下)，我将非常感激。

Answer 1

算法的可能递归版本可能如下所示。

ht_breaks <- function(x){
  ht_inner <- function(x, mu){
    n <- length(x)
    mu <- c(mu, mean(x))
    h <- x[x > mean(x)]
    if(length(h) > 1 && length(h)/n <= 0.4){
      ht_inner(h, mu)
    } else mu
  }
  ht_inner(x, NULL)
}

pareto_data <- (1.0/(1:100))^1.16
ht_breaks(pareto_data)
#[1] 0.03849691 0.17799039 0.48184535