R:使用model.matrix时重命名

Question

运行以下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“。我如何避免这种行为(或者将其重命名，但在非常一般的设置中，例如在用户提供公式的函数中)？

Answer 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"