Как создать переменную на основе количества уникальных значений в другом фрейме данных?

Question

Это упрощенный пример того, что я хочу сделать.

Набор данных 1 (DF1) содержит данные о яблоках (например, размер или количество отверстий), а второй набор данных (DF2) содержит информацию об обнаруженных внутри них червях, включая цвет и в каком яблоке они были обнаружены. Я хочу добавить в DF1 переменную с количеством уникальных цветов (червей), которые существуют в каждом яблоке.

DF1<-data.frame(x=c("A1","A2","A3","A4","A5"),y=c(3,26,5,27,5))
DF2<-data.frame(Q=c("A1","A1","A1","A1","A1","A1","A2","A2","A3","A3","A3","A4","A5","A5","A5","A5"),R=c("red","red","blue","yellow","yellow","blue","orange","orange","green","red","red","blue","blue", "purple","black","red"),S=c(4,5,3,5,4,3,5,4,3,5,4,3,5,4,3,5))

Я новичок в R, и, пытаясь его решить, я подумал о:

DF1$N.Colors<-length(unique(DF2$R[match(DF1$X,DF2$Q)]))

Но он возвращает мне новую переменную, заполненную нулями, вместо желаемого вектора:

 DF1$N.Colors<-c(3,1,2,1,4)

Я был бы очень признателен за вашу помощь с этим

Answer 1

Это можно сделать, используя соединение со столбцами «Q», «x» обоих наборов данных, подсчитав уникальные значения «R» и назначив его новому столбцу в «DF1».

library(data.table)
DF1$N.Colors <- setDT(DF2)[DF1, uniqueN(R), on = .(Q = x), by = .EACHI]$V1

---

Или используя tidyverse

library(dplyr)
DF2 %>%
   group_by(x = Q) %>%
   summarise(N.Colors = n_distinct(R)) %>%
   right_join(DF1)

Answer 2

baseРешение с aggregate()и merge():

merge(DF1, aggregate(N.Colors ~ Q, list(N.Colors = DF2$R, Q = DF2$Q), function(x) length(unique(x))), all.x = T, by.x = "x", by.y = "Q")

#    x  y N.Colors
# 1 A1  3        3
# 2 A2 26        1
# 3 A3  5        2
# 4 A4 27        1
# 5 A5  5        4