如何在R语言中按指定规则排序数据集（x1循环+ x2升序）

需求概述 #

• x1列按2016,2017,2018的循环顺序排列
• x2列按升序排列
• x3列保留与x1、x2的对应关系，无需额外排序

示例数据 #

g <- data.frame("x1" = c(2018,2017,2018,2016,2017,2016,2016,2017,2018),
                "x2" = c(2,2,5,1,4,3,2,1,1),
                "x3" = c(21,22,23,24,25,26,27,28,29))

当前问题 #

现有代码仅实现了x1列的循环排序，但无法同步满足x2列升序的要求：

# step 1
cycle_order <- function(x) {
    order(ave(x, x, FUN=seq_along), x)
}

# step 2 
X <- g[cycle_order(g$x1), ]

解决方案 #

要同时实现x1循环排序和x2升序，需要将两个排序条件结合，先按x2升序分组，再在每组内按x1的循环顺序排列。以下是两种可行方法：

方法一：因子化x1+分组序号排序 #

# 定义x1的目标循环顺序
x1_target_order <- c(2016, 2017, 2018)
# 将x1转为指定顺序的因子，确保排序时遵循该顺序
g$x1_factor <- factor(g$x1, levels = x1_target_order)

# 为每个x2+x1组合分配组内序号，用于实现循环排列
g$group_seq <- ave(g$x1, g$x2, g$x1, FUN = seq_along)

# 按x2升序、组内序号、x1因子顺序排序，得到最终结果
result <- g[order(g$x2, g$group_seq, g$x1_factor), ]

# 移除临时辅助列，保留原数据列
result <- result[, c("x1", "x2", "x3")]

方法二：先按x2排序再分配循环组 #

# 先对数据按x2升序排序
g_sorted_x2 <- g[order(g$x2), ]

# 为每个x1值分配循环批次序号（同一批次的2016、2017、2018属于同一循环组）
g_sorted_x2$cycle_batch <- ave(g_sorted_x2$x1, g_sorted_x2$x1, FUN = seq_along)

# 按循环批次和x1的目标顺序排序
result <- g_sorted_x2[order(g_sorted_x2$cycle_batch, factor(g_sorted_x2$x1, levels = x1_target_order)), ]

# 调整列顺序至原结构
result <- result[, c("x1", "x2", "x3")]

验证结果 #

运行上述任意一种方法，都会得到符合期望的结果：

• x1序列：2016,2017,2018,2016,2017,2018,2016,2017,2018
• x2序列：1,1,1,2,2,2,3,4,5
• x3序列：24,28,29,27,22,21,26,25,23

内容的提问来源于stack exchange，提问作者musa isah