R语言：基于另一数据框均值替换指定变量值的实现需求

Hey there! Since you're new to R programming, let's work through this problem step by step. I'll start with the loop approach you requested, then also share a more efficient R-style method that avoids loops (since we love vectorized operations here).

First: Clarify Data Structure Assumptions #

Before diving into code, let's align on your data setup:

• Let’s call your main 1300-row data frame main_df, with 4 variables: var1, var2, var3, var4. I’m assuming there’s a project code column (like project_id) that links main_df to df_2.
• df_2 stores average values per project: it has the same project code column, plus columns like Avg_var1, Avg_var2, Avg_var3 with the mean values for each variable.

If your df_2 holds global averages (not per project), I’ll share a simplified version too!

Loop Approach (As You Requested) #

Here’s how to implement your logic with a loop. We’ll iterate through each unique project code, grab its averages from df_2, and update values in main_df:

# First, create a backup of your data (always a safe move!)
main_df_backup <- main_df

# Get all unique project codes from your main dataset
unique_projects <- unique(main_df$project_id)

# Loop through each project
for (proj in unique_projects) {
  # Fetch the average values for this project from df_2
  proj_averages <- df_2[df_2$project_id == proj, ]
  
  # Find which rows in main_df belong to this project
  proj_rows <- main_df$project_id == proj
  
  # Update var1: replace values > average with the average, keep others
  main_df$var1[proj_rows] <- ifelse(main_df$var1[proj_rows] > proj_averages$Avg_var1,
                                    proj_averages$Avg_var1,
                                    main_df$var1[proj_rows])
  
  # Repeat for var2
  main_df$var2[proj_rows] <- ifelse(main_df$var2[proj_rows] > proj_averages$Avg_var2,
                                    proj_averages$Avg_var2,
                                    main_df$var2[proj_rows])
  
  # Repeat for var3
  main_df$var3[proj_rows] <- ifelse(main_df$var3[proj_rows] > proj_averages$Avg_var3,
                                    proj_averages$Avg_var3,
                                    main_df$var3[proj_rows])
}

If df_2 Has Global Averages (Not Per Project) #

If df_2 is just a single row with overall averages for var1, var2, var3 (columns like Avg_var1, Avg_var2, Avg_var3), you can loop through the variables instead:

main_df_backup <- main_df

# List of variables to process
target_vars <- c("var1", "var2", "var3")

# Loop through each variable
for (var in target_vars) {
  # Get the corresponding average from df_2
  avg_val <- df_2[[paste0("Avg_", var)]]
  
  # Update the variable: replace values > average with the average
  main_df[[var]] <- ifelse(main_df[[var]] > avg_val, avg_val, main_df[[var]])
}

More Efficient R-Style Method (No Loops!) #

In R, vectorized operations are faster and cleaner than loops (even for 1300 rows). Using the dplyr package (a go-to for data manipulation), here’s how to do it in one concise block:

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

main_df <- main_df %>%
  # Join main_df with df_2 to bring in project-specific averages
  left_join(df_2, by = "project_id") %>%
  # Update each variable with the replacement logic
  mutate(
    var1 = ifelse(var1 > Avg_var1, Avg_var1, var1),
    var2 = ifelse(var2 > Avg_var2, Avg_var2, var2),
    var3 = ifelse(var3 > Avg_var3, Avg_var3, var3)
  ) %>%
  # Remove the temporary average columns
  select(-starts_with("Avg_"))

Quick Tips: #

• Adjust column names (like project_id, Avg_var1) to match your actual data.
• Always backup your data before making changes—you never know when you might need to revert!
• If you’re new to dplyr, the %>% (pipe) operator passes the result of the left side to the right, making code easier to read and follow.

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