R语言：基于循环生成的未知变量名向量创建求和新列

Got it, let's walk through how to solve this problem in R—since your dataset and example look like R code, I'll focus on that. The key here is handling the dynamic number of variable vectors and automating column creation without hardcoding each case.

Step 1: Set Up Your Example Data & Vectors #

First, let's replicate your sample data and the variable vectors you mentioned:

# Create the sample dataset
data <- data.frame(
  Name = c("r1", "r2", "r3", "r4"),
  A = c(1, 2, 5, 7),
  B = c(5, 4, 15, 8),
  C = c(12, 7, 6, 0),
  D = c(21, 10, 9, 7),
  E = c(15, 9, 6, 18),
  row.names = c("r1", "r2", "r3", "r4")
)

# Define your variable vectors (these could be generated dynamically in your loop)
Vec1 <- c("A", "B", "C")
Vec2 <- c("B", "D")
Vec3 <- c("D", "E")

# Store all vectors in a list—this lets us handle any number of vectors easily
vec_collection <- list(Vec1, Vec2, Vec3)

Step 2: Dynamically Generate Sum Columns #

Now we'll loop through each vector in the list, create the concatenated column name, and calculate the row-wise sum for the specified variables:

# Loop through each vector in our collection
for(var_vec in vec_collection) {
  # Create the new column name by joining the variable names
  new_col <- paste(var_vec, collapse = "")
  
  # Calculate row-wise sum of the selected columns and add to the dataset
  data[[new_col]] <- rowSums(data[var_vec])
}

Step 3: Verify the Result #

If you print the updated data frame, you'll get exactly the output you expected:

print(data)

Output: #

Name A  B  C ABC  D BD  E DE
r1  r1  1  5 12  18 21 26 15 36
r2  r2  2  4  7  13 10 14  9 19
r3  r3  5 15  6  26  9 24  6 15
r4  r4  7  8  0  15  7 15 18 25

Optional: Add Error Checking #

If there's a chance your dynamic vectors might include variable names that don't exist in the dataset, you can add a check to avoid errors:

for(var_vec in vec_collection) {
  # Check if all variables in the vector exist in the dataset
  if(all(var_vec %in% colnames(data))) {
    new_col <- paste(var_vec, collapse = "")
    data[[new_col]] <- rowSums(data[var_vec])
  } else {
    # Warn about missing variables instead of crashing
    missing_vars <- setdiff(var_vec, colnames(data))
    warning(paste("Skipping vector: Variables", paste(missing_vars, collapse = ", "), "not found in the dataset"))
  }
}

This approach works no matter how many variable vectors you generate—just add them to the vec_collection list, and the loop will handle the rest.

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