有沒有一種方法可以將回歸中的重要變量放入圖表中？

Question

以mtcars為例：

> reg <- lm(mpg ~ cyl + disp + hp + drat + wt, data = mtcars)
> sigvar <- data.frame(summary(reg)$coef[summary(reg)$coef[,4] <= .05, 4]) #extracts significant variables with p-values
> rownames <- rownames(sig) #extracts the variables only

我希望將行名放在小節的x軸上，小節的高度將是所述行名的平均值。 謝謝。

Answer 1

我不清楚您希望條形圖的高度位於條形圖中。 然而：

reg = lm(mpg ~ cyl + disp + hp + drat + wt, data = mtcars)
## extracts significant variables with p-values
## note the -1 means you skip the intercept 
sigvar = summary(reg)$coef[-1,4] <= .05 

如果我了解您想要的內容，那么您想要一個條形圖，這些條形圖的平均值與條形的高度一樣重要。 您需要將有效變量與數據框中的變量名稱進行匹配

i = match(names(sigvar)[sigvar], names(mtcars))

i現在包含原始數據幀中與有效變量相對應的列。 不幸的是，對於mtcars數據，這意味着mtcars[,i]僅返回一列，因此通常我會做類似

barplot(sapply(mtcars[,i], mean))

但這在這里做不正確的事情，因為mtcars[,i]返回一個向量。 讓我們假設i = c(5,6) ，那么這將起作用

i = 5:6
barplot(sapply(mtcars[,i], mean))