如何用R语言列表简化重复代码？以mtcars数据过滤为例

Awesome question—this is exactly the kind of scenario where lists and functional programming tools eliminate redundant code and make your workflow way more efficient. Let’s break down the optimal approach step by step, using tidyverse tools since you’re already working with dplyr.

Step 1: Reduce Redundancy with a Base Filtered Subset #

First, let’s avoid repeating the gear == 4 and date == '2005' filter conditions by creating a single subset upfront. This cuts down on duplicate code and makes your workflow cleaner:

library(tidyverse)

# Create a base subset of 2005, gear=4 data
filtered_base <- data %>% 
  filter(gear == 4, date == '2005')

Step 2: Generate Rank Intervals with seq() #

Instead of hardcoding each rank range, use seq() to dynamically create your intervals. We’ll pair it with map2() to generate the exact groups you need (1-5, 6-10, 11-15):

# Generate start and end points for each rank group
start_ranks <- seq(1, 11, by = 5)
end_ranks <- seq(5, 15, by = 5)

# Create a list of rank intervals
rank_groups <- map2(start_ranks, end_ranks, ~.x:.y)

# Optional: Name the list for clarity (makes later steps easier)
names(rank_groups) <- str_c("Rank_", start_ranks, "-", end_ranks)

Step 3: Filter & Store Results in a List #

Now use map() to apply the rank filter to each interval in rank_groups, storing the results in a list. This replaces your three separate filter calls with one line of code:

# Apply filter to each rank interval and store in a list
rank_subsets <- rank_groups %>% 
  map(~filtered_base %>% filter(rank %in% .x))

Step 4: Plot with ggplot2 #

You have two great options for plotting, depending on whether you want combined visuals or separate plots:

Option 1: Combine into One Data Frame for Faceted Plots #

Bind the list into a single data frame with a grouping column, then use facet_wrap() to compare groups side-by-side:

# Combine list into a single data frame with group labels
combined_data <- rank_subsets %>% 
  bind_rows(.id = "rank_group")

# Create faceted plot
ggplot(combined_data, aes(x = mpg, y = wt)) +
  geom_point(color = "#2c3e50", alpha = 0.7) +
  facet_wrap(~rank_group) +
  labs(title = "2005 Gear=4: MPG Rank Groups",
       x = "Miles Per Gallon",
       y = "Vehicle Weight") +
  theme_minimal()

Option 2: Generate Individual Plots & Combine #

If you want separate plots for each rank group, use imap() to create ggplot objects for each subset, then combine them with the patchwork package:

# Install patchwork if you haven't already
# install.packages("patchwork")
library(patchwork)

# Create a list of ggplot objects
rank_plots <- rank_subsets %>% 
  imap(function(df, group_name) {
    ggplot(df, aes(x = mpg, y = hp)) +
      geom_point(color = "#e74c3c") +
      labs(title = group_name,
           x = "MPG",
           y = "Horsepower") +
      theme_light()
  })

# Combine plots into a grid
rank_plots[[1]] + rank_plots[[2]] + rank_plots[[3]] +
  plot_layout(ncol = 2) +
  plot_annotation(title = "2005 Gear=4: Rank Group Comparisons")

Why This Works #

• Less redundancy: You only write the filter logic once, even if you add more rank groups later (just adjust the seq() parameters!).
• Flexibility: Lists let you easily store, manipulate, and iterate over multiple data subsets.
• Scalability: If you ever need to expand to more years or gear types, you can wrap this logic in another map() call to handle multiple groups at once.

Base R Alternative (No Tidyverse) #

If you prefer base R, you can use lapply() instead of map():

# Base R version
rank_groups <- list(c(1:5), c(6:10), c(11:15))
names(rank_groups) <- c("Rank_1-5", "Rank_6-10", "Rank_11-15")

rank_subsets <- lapply(rank_groups, function(r) {
  subset(filtered_base, rank %in% r)
})

内容的提问来源于stack exchange，提问作者datanalyst