有没有直接的方法来交换 R 中的列？

Question

我想知道是否有一种直接的方法可以在R 中交换matrix 的任何一对列。

假设我们要交换矩阵中的第一列 (i=1) 和第三列 (j=3)

n=3
mat  <- matrix(1:(n^2),ncol=n)

     [,1] [,2] [,3]
[1,]    1    4    7
[2,]    2    5    8
[3,]    3    6    9

无论矩阵的大小n如何，以下始终有效

i=1; j=3
mat2     <- mat[,i]
mat[,i]  <- mat[,j]
mat[,j]  <- mat2

而像cbind(mat[,3],mat[,2],mat[,1])（在早期线程SO1 和SO2 中建议）这样的直接命令仅限于给定的n。

有没有更直接的方法来交换大小为n 的矩阵中的两列i 和j？

Answer 1

可以使用旋转索引来完成列旋转。假设您有矩阵 mat 和 n 列，并且您想要旋转列 i 和 j，您可以这样做：

piv <- 1:n; piv[c(i,j)] <- as.integer(c(j,i))
mat <- mat[, piv]

as.integer 调用是为了提高内存效率。 piv <- 1:n 是一个整数向量，如果 i 和 j 是双精度数，只需调用 piv[c(i,j)] <- c(j,i) 将导致 piv 首先被复制到双精度数向量中，然后被更新。