5.4 Examine the data

In Chapter 3, we learned how to numerically and visually summarize variables one at a time. Before fitting a linear regression model, it is a good idea to look at all the variables individually. At this point, we are looking for very obvious problems, such as values that are highly unusual, extremely skewed variables, or categorical variables with sparse levels. We also want to check that all categorical variables are coded as factors.

5.4.1 Outcome

We need to make sure the outcome variable is appropriate for a linear regression. It should be continuous (not just a few unique values) and, ideally, normally distributed (or at least symmetric), although we will learn that normality is not essential if the sample size is large.

To be sure the outcome really is continuous, count the number of unique values. If there are only a few, then linear regression may not be appropriate; consider instead ordinal logistic regression (Section 6.22) or binary logistic regression (after dichotomizing the outcome) (Chapter 6). Next, create a numeric summary and plot a histogram of the outcome. Later, after you have learned more about regression diagnostics, you will be able to identify skew and outliers in these plots and may immediately realize that a transformation is needed. You will investigate these potential problems more closely when checking the model assumptions in Sections 5.14 to 5.17, but sometimes you can save time by noticing the need for a transformation before fitting the model.

Example 5.1 (continued): Examine a numeric summary and histogram of the outcome, fasting glucose.

# Count the number of unique values
## [1] 96
# Numerical summary
##    Min. 1st Qu.  Median    Mean 3rd Qu.    Max. 
##    2.61    5.33    5.72    6.11    6.22   19.00
# Histogram
hist(nhanesf.complete$LBDGLUSI, main = "", xlab = "Fasting Glucose (mmol/L)")
Histogram of Fasting Glucose (mmol/L)

Figure 5.1: Histogram of Fasting Glucose (mmol/L)

The summary indicates that the outcome has many (96) unique values, so it is appropriate to treat it as continuous. While there are no strange values, Figure 5.1 shows that the outcome is skewed.

5.4.2 Continuous predictors

Count the number of unique values – if there are only a few then you should consider treating the predictor as categorical instead. Create a numeric summary and plot a histogram for each continuous predictor. There is no requirement that continuous predictors be normally distributed, however skewed distributions and extreme values can lead to a problem with influential observations, which you will learn more about in Section 5.22. What you are primarily looking for at this point are strange values and to be sure each predictor is coded as you expect. For example, if what you thought should be a continuous predictor has only a few unique values, then perhaps the variable should be coded as categorical instead.

Example 5.1 (continued):

# Count the number of unique values
## [1] 375
## [1] 61
# Numerical summary
##    Min. 1st Qu.  Median    Mean 3rd Qu.    Max. 
##    63.2    88.3    98.3   100.8   112.2   169.5
##    Min. 1st Qu.  Median    Mean 3rd Qu.    Max. 
##    20.0    34.0    47.0    47.8    61.0    80.0
# Histograms
hist(nhanesf.complete$BMXWAIST, main = "", xlab = "Waist Circumference (cm)")
hist(nhanesf.complete$RIDAGEYR, main = "", xlab = "Age (years)")
Histograms of continuous predictors

Figure 5.2: Histograms of continuous predictors

The summaries indicate the following:

  • Each predictor has many unique values indicating that each should indeed be treated as continuous.
  • Figure 5.2 shows that waist circumference is slightly skewed, but neither waist circumference nor age have very extreme values. We will still, however, check for influential observations in Section 5.22 since these histograms can only show extreme values one-variable-at a time – what matters for MLR is how different a case is from the others when considering all the predictors simultaneously.
  • Neither variable has any strange values

What if there were strange values? For example, if there were age values of, say, 142 or -5, then we should investigate further. Are these data entry errors that can be corrected? If they cannot be corrected with certainty, then it is reasonable to set these values to missing.

5.4.3 Categorical predictors

The main issue to watch out for with categorical predictors is sparse levels (levels that have a small sample size) since any estimate based on a sparse level will be very imprecise. Create a frequency table for each categorical predictor (optionally, also plot a bar chart). Look for levels that have only a few observations as these may need to be collapsed with other levels. You could use, for example, 30, 50, or 100 as a cutoff, but any cutoff is subjective. If you are concerned that your regression results may depend too much on the choice of how to collapse categorical predictors, carry out a sensitivity analysis to compare the results under different scenarios (see Section 5.25). Finally, check that each categorical predictor is coded as a factor, recoding as a factor if not (see Section 4.4.1).

Example 5.1 (continued):

# Categorical predictors
table(nhanesf.complete$smoker, exclude = NULL)
##   Never    Past Current 
##     500     222     135
table(nhanesf.complete$RIAGENDR, exclude = NULL)
##   Male Female 
##    388    469
# table(nhanesf.complete$RIDRETH3, exclude = NULL)
# Use count so the display is vertical
# (easier to read when there are many levels)
nhanesf.complete %>%
##             RIDRETH3   n
## 1   Mexican American  97
## 2     Other Hispanic  55
## 3 Non-Hispanic White 534
## 4 Non-Hispanic Black  98
## 5 Non-Hispanic Asian  36
## 6        Other/Multi  37
table(nhanesf.complete$income, exclude = NULL)
##            < $25,000 $25,000 to < $55,000             $55,000+ 
##                  157                  221                  479

From the output we conclude the following:

  • Each predictor has only a few levels, so treating each as categorical is appropriate.
  • The race/ethnicity variable may have a few sparse levels.
  • Each is coded as a factor.

5.4.4 Overall description of the data

As a partial alternative to the steps described above, you can use Hmisc::describe() (see Section 3.1.1) to get a detailed description of all the variables with one command. It does not provide all the information we need (e.g., whether or not a categorical variable is coded as a factor), but it is useful.

nhanesf.complete %>% 
         RIAGENDR, RIDRETH3, income) %>% 
# (results not shown)

5.4.5 Collapsing sparse levels

This section demonstrates how to collapse a categorical predictor that has sparse levels. Remember that the cutoff for “sparse” is arbitrary. Here, we combine race/ethnicity levels that have about 55 cases or fewer with other levels to create a new variable. By this definition, the “Mexican American” group is sparse and it makes sense to combine it with the “Hispanic” group. Additionally, “Non-Hispanic Asian” and “Other/Multi” are both small, so we combine them. Remember to always check any derivation by comparing observations before and after.

nhanesf.complete <- nhanesf.complete %>% 
  mutate(race_eth = fct_collapse(RIDRETH3,
   "Hispanic"           = c("Mexican American", "Other Hispanic"),
   "Non-Hispanic Other" = c("Non-Hispanic Asian", "Other/Multi")))

# Check derivation
TAB <- table(nhanesf.complete$RIDRETH3, nhanesf.complete$race_eth, exclude = NULL)
addmargins(TAB, 1) # Adds a row of column totals
##                      Hispanic Non-Hispanic White Non-Hispanic Black Non-Hispanic Other
##   Mexican American         97                  0                  0                  0
##   Other Hispanic           55                  0                  0                  0
##   Non-Hispanic White        0                534                  0                  0
##   Non-Hispanic Black        0                  0                 98                  0
##   Non-Hispanic Asian        0                  0                  0                 36
##   Other/Multi               0                  0                  0                 37
##   Sum                     152                534                 98                 73

The two-way frequency table shows that the derivation code did what we intended (“Mexican American” and “Other Hispanic” were combined into “Hispanic”; “Non-Hispanic Asian” and “Other/Multi” were combined into “Non-Hispanic Other”). The “Sum” row provides the sample sizes in the new levels.

NOTE: This example is provided to illustrate how to go about collapsing a sparse predictor. In practice, think about the subject matter carefully. By collapsing the way we did, we are assuming that the outcome, fasting glucose, is not meaningfully different between the groups of individuals in the levels that were combined. An alternative would be to exclude individuals in sparse levels from an analysis entirely, which would result in altering the scope of the analysis. For example, if we excluded those in the “Other/Multi” group, the results would not be generalizable to that group.

5.4.6 Visualizing the unadjusted relationships

Before fitting the model, visualize the association between the outcome and each predictor. These plots will not correspond exactly to the MLR model since they will not be adjusted for any other predictors, but they are a useful place to start just to make sure the predictors you are using are what you think they are (if the plot looks very strange, go back and look at your code to make sure you are using the variables you want to use and that they were coded correctly). Later, in Section 5.7, we will examine plots that are adjusted for the other predictors in the model.

Example 5.1 (continued): Plot the relationship between fasting glucose (LBDGLUSI) and each of waist circumference (BMXWAIST), smoking status (smoker), age (RIDAGEYR), gender (RIAGENDR), race/ethnicity (RIDRETH3), and income, one predictor at a time. The plots are shown in Figure 5.3.

You could use the code found in Section 4.1 to create each plot one at a time. Or, you can use the function plotyx() found in Functions_rmph.R (which you loaded at the beginning of this chapter). When you are faced with repeating code multiple times with only minor changes, you can make your code script easier to read (and avoid cut-and-paste errors) by using a function. Each time you call this function, pass the names of the specific outcome and predictor you are interested in to the function as arguments. Before plotting, the code below makes copies of the race/ethnicity and income variables with shorter labels so they will fit in the plot axis.

# Rename the levels with shorter labels that fit in the plot
## [1] "Hispanic"           "Non-Hispanic White" "Non-Hispanic Black" "Non-Hispanic Other"
nhanesf.complete$race_ethb <- nhanesf.complete$race_eth
levels(nhanesf.complete$race_ethb) <- c("H", "NHW", "NHB", "NHO")

## [1] "< $25,000"            "$25,000 to < $55,000" "$55,000+"
nhanesf.complete$incomeb <- nhanesf.complete$income
levels(nhanesf.complete$incomeb) <- c("<$25K", "$25K-<$55K", "$55K+")

NOTE: The syntax is plotyx("y", "x", data) (outcome first, then predictor, then dataset) where "y" and "x" are character strings naming the variables you want to plot.

# If you have not already loaded this file...
# ...do so now to make contplot() and catplot() available

# To change margin sizes
# Default par(mar=c(5, 4, 4, 2) + 0.1) for c(bottom, left, top, right)
par(mar=c(4.5, 4, 1, 1))
plotyx("LBDGLUSI", "BMXWAIST",  nhanesf.complete,
       ylab = "FG (mmol/L)", xlab = "Waist Circumference (cm)")
plotyx("LBDGLUSI", "smoker",    nhanesf.complete,
       ylab = "FG (mmol/L)", xlab = "Smoking Status")
plotyx("LBDGLUSI", "RIDAGEYR",  nhanesf.complete, 
       ylab = "FG (mmol/L)", xlab = "Age (years)")
plotyx("LBDGLUSI", "RIAGENDR",  nhanesf.complete, 
       ylab = "FG (mmol/L)", xlab = "Gender")
# Using race_ethb and incomeb (the versions with shorter labels)
plotyx("LBDGLUSI", "race_ethb", nhanesf.complete, 
       ylab = "FG (mmol/L)", xlab = "Race/Ethnicity")
plotyx("LBDGLUSI", "incomeb",   nhanesf.complete,
       ylab = "FG (mmol/L)", xlab = "Income")
Unadjusted associations with fasting glucose (mmol/L)

Figure 5.3: Unadjusted associations with fasting glucose (mmol/L)