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"