一次组合R中的许多列

Question

我有一个类似于以下的数据框：

A1U_sweet  A2F_dip  A3U_bbq  C1U_sweet  C2F_dip  C3U_bbq
1          2        1        NA         NA       NA
NA         NA       NA       4          1        2
2          4        7        NA         NA       NA

我想创建额外的列来组合 A 值和 C 值。生成的数据框将包含看起来像 B1U_sweet 和 B2F_dip 的列。

A1U_sweet  A2F_dip  A3U_bbq  C1U_sweet  C2F_dip  C3U_bbq  B1U_sweet  B2F_dip
1          2        1        NA         NA       NA       1          2
NA         NA       NA       4          1        2        4          1
2          4        7        NA         NA       NA       2          4

有没有一种方法可以让我创建这些额外的列，而无需将 A 列和 C 列一一合并？在过去，我将它们单独组合在一起，但如果有类似循环的东西匹配列 A1 和 C1、A2 和 C2 等，那就太好了。有 20 个“A”列和 20 个“C”列我想合并成“B”列。

编辑：解决方案是这里公认的答案：Combining columns in R based on matching beginnings of column title names

Answer 1

你可以这样做：

types <- grep('^A', names(df)) ## Get all "A" patterns
types <- substr(types, 2, Inf) ## Remove the "A"
for (tp in types) {
  aa <- df[[paste0('A', tp)]] ## "A" column
  cc <- df[[paste0('C', tp)]] ## "C" column
  df[[paste0('B', tp)]] <- ifelse(is.na(aa), aa, cc)
}

编辑：如果您的 df 中的其他一些列以 A 开头，您可以尝试以下 3 件事之一：

1. 首先从 df 中删除它们

2. 不要 grep 类型，而是手动输入它们（例如 types <- c("1U_sweet", ...)

3. 如果您的类型遵循特定模式，您可以将 grep 中的正则表达式更改为 "^A[0-9][0-9][A-Z]_[a-z]+"

Answer 2

不使用循环的 dplyr 方法：

df = read.table(text = "
A1U_sweet  A2F_dip  A3U_bbq  C1U_sweet  C2F_dip  C3U_bbq
1          2        1        NA         NA       NA
NA         NA       NA       4          1        2
2          4        7        NA         NA       NA
", header=T)

library(tidyverse)

df %>%
  mutate(row_id = row_number()) %>%
  gather(col,value,-row_id) %>%
  mutate(col1 = substr(col, 1, 1),
         col2 = substr(col, 2, nchar(col))) %>%
  group_by(row_id, col2) %>%
  summarise(new_value = unique(value[!is.na(value)])) %>%
  ungroup() %>%
  mutate(col2 = paste0("B",col2)) %>%
  spread(col2,new_value) %>%
  bind_cols(df) %>%
  select(-row_id, -B3U_bbq)

# # A tibble: 3 x 8
#   B1U_sweet B2F_dip A1U_sweet A2F_dip A3U_bbq C1U_sweet C2F_dip C3U_bbq
#       <int>   <int>     <int>   <int>   <int>     <int>   <int>   <int>
# 1         1       2         1       2       1        NA      NA      NA
# 2         4       1        NA      NA      NA         4       1       2
# 3         2       4         2       4       7        NA      NA      NA