R语言按指定type顺序排序数据行并保留cat列相对顺序的方法

初始数据与需求 #

构造测试数据集的代码如下：

dat <- data.frame(
  type = c("Y","Y","Y","Y","X","X","X","X","Z","Z","Z","Z"), 
  cat = c("A", "B", "C", "D","A", "B", "C", "D","A", "B", "C", "D"), 
  id = c(1,2,3,4,5,6,7,8,9,10,11,12)
)

原始数据预览：

type cat id
1     Y   A  1
2     Y   B  2
3     Y   C  3
4     Y   D  4
5     X   A  5
6     X   B  6
7     X   C  7
8     X   D  8
9     Z   A  9
10    Z   B 10
11    Z   C 11
12    Z   D 12

核心需求：将数据行按type列的自定义顺序x <- c("Z", "X", "Y")重排，同时每个type分组内保留cat列原有的A→B→C→D相对顺序。

原有代码的错误原因 #

• 第一次使用slice(match(x, type))的问题：match()函数对每个匹配值只会返回第一个命中的行索引，最终只能取出3行数据，无法保留全部12行。
• 第二次循环构造匹配向量的问题：match()匹配重复值时始终返回第一个匹配项的位置，因此重复的type值只会匹配到每个type分组下第一行（即cat=A的行），最终返回结果所有cat都为A。

正确实现方式 #

核心思路是将type列转换为自定义层级的因子，排序时会自动按照因子层级顺序排列，同层级内保留原始行的相对顺序，正好满足需求。

dplyr实现（最常用） #

library(dplyr)
x <- c("Z", "X", "Y")
res <- dat %>%
  mutate(type = factor(type, levels = x)) %>%
  arrange(type)

运行结果符合预期：

type cat id
1     Z   A  9
2     Z   B 10
3     Z   C 11
4     Z   D 12
5     X   A  5
6     X   B  6
7     X   C  7
8     X   D  8
9     Y   A  1
10    Y   B  2
11    Y   C  3
12    Y   D  4

基础R实现（无依赖） #

不需要加载第三方包，直接用基础排序逻辑即可：

x <- c("Z", "X", "Y")
dat$type <- factor(dat$type, levels = x)
res <- dat[order(dat$type), ]

data.table实现（高性能） #

适合大数据量场景，排序速度更快：

library(data.table)
setDT(dat)
x <- c("Z", "X", "Y")
dat[, type := factor(type, levels = x)]
setorder(dat, type)

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