如何基于无重复元素的指定向量对R数据框进行排序

嘿，这个需求我经常碰到，用R的话可以通过**因子（factor）**轻松实现自定义顺序的排序，下面一步步来演示：

首先咱们先还原你的示例数据：

# 创建示例数据框
df <- data.frame(
  col1 = c("A", "B", "C", "D", "E", "F", "G", "H"),
  col2 = c("a", "b", "b", "b", "a", "a", "c", "b")
)

# 你的自定义排序向量
custom_order <- c("b", "c", "a")

方法一：修改列类型后排序 #

把需要排序的列转换为因子，并且指定levels参数为你的自定义向量——这一步就是告诉R：我要按这个顺序来排！

# 将col2转为因子，指定自定义顺序
df$col2 <- factor(df$col2, levels = custom_order)

# 按col2排序
sorted_df <- df[order(df$col2), ]

方法二：不修改原数据框，直接排序 #

如果你不想改动原数据框的结构，可以在排序时直接转换因子：

sorted_df <- df[order(factor(df$col2, levels = custom_order)), ]

验证结果 #

运行上面的代码后，sorted_df的输出就是你想要的结果：

col1 col2
2    B    b
3    C    b
4    D    b
8    H    b
7    G    c
1    A    a
5    E    a
6    F    a

小提醒 #

如果你的目标列里存在不在custom_order中的元素，默认会被排在所有指定元素的后面。要是想过滤掉这些元素，可以提前用df <- df[df$col2 %in% custom_order, ]处理。

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