R：使用 model.matrix 时重命名

【问题标题】：R: Renaming when using model.matrixR：使用 model.matrix 时重命名
【发布时间】：2016-03-21 10:59:14
【问题描述】：

运行以下 R 代码时：

y <- rnorm(100)  
B <- matrix(rnorm(200), ncol=2)  
colnames(B) <- c("x1", "x2")  
A <- model.matrix(y~B)  
colnames(A)

model.matrix 将矩阵 A 中的前缀“B”添加到变量名称中。我怎样才能避免这种行为（或在非常一般的设置中重命名它，例如在具有用户提供公式的函数中）？

【问题讨论】：

【解决方案1】：

我们可以使用sub 删除列名中的第一个字符。

colnames(A)[-1] <- sub("^.", "", colnames(A)[-1])

或者通过连接来自“B”的列名来设置列名

colnames(A) <- c(colnames(A)[1], colnames(B))

不使用sub 的另一种选择是使用“y”和“B”矩阵创建一个data.frame，然后使用y ~.

A <- model.matrix(y~., data=data.frame(y, B))
colnames(A)
#[1] "(Intercept)" "x1"          "x2"

【讨论】：