R Loop for Variable Names to run linear regression model

by

Ok, I’ll post an answer. I will use the dataset mtcarsas an example. I believe it will work with your dataset.
First, I create a store, lm.test, an object of class list. In your code you are assigning the output of lm(.) every time through the loop and in the end you would only have the last one, all others would have been rewriten by the newer ones.
Then, inside the loop, I use function reformulate to put together the regression formula. There are other ways of doing this but this one is simple.

# Use just some columns
data <- mtcars[, c("mpg", "cyl", "disp", "hp", "drat", "wt")]
col10 <- names(data)[-1]

lm.test <- vector("list", length(col10))

for(i in seq_along(col10)){
    lm.test[[i]] <- lm(reformulate(col10[i], "mpg"), data = data)
}

lm.test

Now you can use the results list for all sorts of things. I suggest you start using lapply and friends for that.
For instance, to extract the coefficients:

cfs <- lapply(lm.test, coef)

In order to get the summaries:

smry <- lapply(lm.test, summary)

It becomes very simple once you’re familiar with *apply functions.

More Related Contents:

  1. How to Loop/Repeat a Linear Regression in R
  2. Loop through a R matrix row-wise and print elements [duplicate]
  3. Is the “*apply” family really not vectorized?
  4. Loop in R to read many files
  5. Fitting a linear model with multiple LHS
  6. How to force R to use a specified factor level as reference in a regression?
  7. `lm` summary not display all factor levels
  8. How does predict.lm() compute confidence interval and prediction interval?
  9. Create group number for contiguous runs of equal values
  10. Merge several data.frames into one data.frame with a loop
  11. Remove/collapse consecutive duplicate values in sequence
  12. Is there a better alternative than string manipulation to programmatically build formulas?
  13. Merge by Range in R – Applying Loops
  14. Linear Regression with a known fixed intercept in R
  15. Which is the best method to apply a script repetitively to n .csv files in R?
  16. Interpretation of ordered and non-ordered factors, vs. numerical predictors in model summary
  17. Comparing two vectors in an if statement
  18. lme4::lmer reports “fixed-effect model matrix is rank deficient”, do I need a fix and how to?
  19. predict.lm() in a loop. warning: prediction from a rank-deficient fit may be misleading
  20. How `poly()` generates orthogonal polynomials? How to understand the “coefs” returned?
  21. lm function in R does not give coefficients for all factor levels in categorical data
  22. R assigning ggplot objects to list in loop
  23. how to create a loop that includes both a code chunk and text with knitr in R
  24. Create a loop that includes both a code chunk and text
  25. predict.lm() with an unknown factor level in test data
  26. Regression (logistic) in R: Finding x value (predictor) for a particular y value (outcome)
  27. Messy plot when plotting predictions of a polynomial regression using lm() in R
  28. how to debug “factor has new levels” error for linear model and prediction [duplicate]
  29. Error in Confusion Matrix : the data and reference factors must have the same number of levels
  30. How to use data.table within functions and loops?

Leave a Comment