R语言如何为order函数传入自定义比较函数实现向量自定义排序

问题解答 #

R 基础包的 order() 函数确实没有直接传入自定义比较函数的参数，但针对你的场景有非常适配的原生实现方案，无需引入新类、也无需处理字符串转换的兼容问题： #

方案1：多列降序排序（最适配你的场景）

你的排序逻辑本质就是按向量的第1位降序排列，第1位相同则按第2位降序排列，以此类推，完全可以直接用 order() 的多参数特性实现：

# 把列表中的向量按行拼接为矩阵，行名对应原列表的键
score_mat <- do.call(rbind, scores)
# 将矩阵每一列作为order的排序参数，统一指定降序规则
sorted_idx <- do.call(order, c(as.data.frame(score_mat), decreasing = TRUE))
# 提取排序后的键
sorted_names <- rownames(score_mat)[sorted_idx]

运行后得到的 sorted_names 就是你预期的 d, b, a, c。该方案完全基于R基础函数，对大于9的数值没有任何兼容问题，排序效率远高于自定义R层比较函数。 #

方案2：通用自定义比较规则实现

如果后续有更复杂的排序逻辑无法转换为多列排序，可以通过生成排序权重的方式实现，无需定义S3类：

# 为每个向量生成唯一排序权重，权重大小符合你的比较规则
max_val <- max(unlist(scores))
base <- max_val + 1 # 基数取单元素最大值+1，保证前位优先级高于所有后位之和
xtfrm_scores <- sapply(scores, function(x) {
  sum(x * base^(rev(seq_along(x)) - 1))
})
sorted_names <- names(sort(-xtfrm_scores))

该方案通过指数加权的方式把向量映射为唯一数值，完全保留你需要的「首个差异位决定排序优先级」的规则，也不存在数值溢出或字符串匹配的问题。 #

内容的提问来源于stack exchange，提问作者Felix Jassler