根据 R 中两个变量的唯一值创建数据框

Question

我想根据来自另一个数据框的两个唯一值创建一个新数据框。

id <- c("A", "B", "B", "C")
st.name <- c("TX", "TX", "CA", "CA") 
type <- c(21, 26, 29, 24)
DF <- data.frame(id, st.name, type)
print(DF)

id st.name type  
A      TX   21  
B      TX   26  
B      CA   29  
C      CA   24  

我想根据 id 和 st.type 的唯一值创建一个新的数据框。结果如下所示：

new_id <- c("ATX", "ACA", "BTX", "BCA", "CTX", "CCA")
new_type <- c(21, NA, 26, 29, NA, 24)
DF2 <- data.frame(new_id, new_type)
print(DF2)

new_id new_type  
ATX       21  
ACA       NA  
BTX       26  
BCA       29  
CTX       NA  
CCA       24  

我在以前的项目中使用过dcast，但我不确定如何在此处合并该功能。

Answer 1

我们可以使用来自tidyr 的complete 和unite 来做到这一点

library(tidyr)
complete(DF, id, st.name) %>% 
              unite(new_id, id, st.name, sep = '')

---

或者使用base R，我们可以通过expand.grid、merge和paste得到预期的输出

transform(merge(expand.grid(lapply(DF[1:2], unique)),
             DF, all.x=TRUE), id = paste0(id, st.name))[-2]

---

或data.table 选项和CJ（@Frank 在 cmets 中提到）

library(data.table)
setDT(DF)[CJ(id = id, st.name = st.name, unique=TRUE), on=.(id, st.name)]