如何在data.table中按指定模式重排列，分组同国家的PSRS_与AR_列

核心思路 #

• 首先将所有列划分为三类：固定首列Week、中间国家关联列、固定尾列totalPSRS与totalAR
• 对中间的国家关联列，拆分出「国家名」和「前缀（PSRS/AR）」两个排序键，先按国家名字母序排序，同国家下前缀为PSRS的列排在AR列前面
• 用data.table内置的setcolorder函数修改列顺序，属于原地修改，不需要复制整个表，效率更高

实现代码 #

# 提取全部列名
all_cols <- names(test)

# 定义固定位置的列
first_col <- "Week"
last_cols <- c("totalPSRS", "totalAR")

# 筛选出所有国家关联列（排除固定首尾列）
country_cols <- setdiff(all_cols, c(first_col, last_cols))

# 拆分出国家名和前缀，将前缀转为因子控制排序优先级
country_name <- sub("^(PSRS|AR)_", "", country_cols)
prefix <- factor(sub("_.*$", "", country_cols), levels = c("PSRS", "AR"))

# 按规则排序国家关联列
sorted_country_cols <- country_cols[order(country_name, prefix)]

# 拼接最终列顺序并应用到data.table
final_col_order <- c(first_col, sorted_country_cols, last_cols)
setcolorder(test, final_col_order)

执行完上述代码后，打印names(test)即可验证列顺序已经符合要求。

原代码错误说明 #

你之前编写的代码是对data.table的行进行排序，并不是修改列顺序，因此无法得到预期效果。data.table的x[i,j,by]语法中，第一个位置的参数仅控制行的筛选和排序，修改列顺序需要使用专门的setcolorder函数，或是直接对names(test)重赋值。

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