按指定向量顺序筛选数据框中匹配列名的列

我明白你的需求：你想根据字符串向量筛选数据框的列，但希望结果列的顺序和向量的原始顺序一致，而不是默认的字母序。

问题根源 #

你当前用which(colnames(df) %in% v)来获取列索引，这个方法返回的是数据框中匹配列的原始位置，而这些位置是按数据框列的字母顺序排列的，所以最终筛选后的列自然是字母序。

解决方案 #

直接使用你的向量v（或过滤后确保存在于列名的版本）作为列选择的索引即可，R会严格按照向量的顺序提取列：

# 原始数据框和向量
df <- head(t(USArrests))
v <- c("Wyoming", "Alabama", "California", "Arizona")

# 可选：先过滤掉向量中不在数据框列名里的元素（避免报错）
filtered_v <- v[v %in% colnames(df)]

# 按向量原始顺序筛选列
result_df <- df[, filtered_v]

# 查看结果
print(result_df)

运行结果 #

运行后你会得到符合预期的输出：

Wyoming Alabama California Arizona
Murder         6.8    13.2        9.0     8.1
Assault      161.0   236.0      276.0   294.0
UrbanPop      60.0    58.0       91.0    80.0
Rape          15.6    21.2       40.6    31.0

补充说明 #

如果你的向量v里所有元素都肯定存在于数据框的列名中，那可以直接简化为df[, v]，不需要额外过滤。但加上过滤步骤能避免因向量中存在无效列名而引发的报错，让代码更健壮。

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