Como criar uma variável com base no número de valores únicos em outro quadro de dados?

Question

Este é um exemplo simplificado do que eu quero fazer.

O conjunto de dados 1 (DF1) contém dados de maçãs (como tamanho ou número de buracos) e um segundo conjunto de dados (DF2) contém informações de vermes encontrados dentro delas, incluindo cor e em qual maçã foram encontrados. O que eu quero fazer é adicionar uma variável no DF1 com o número de cores únicas (dos vermes) que existem em cada maçã.

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))

Sou novo no R, e ao tentar resolver pensei em:

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

Mas me devolve uma nova variável preenchida com 0s, em vez do vetor desejado:

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

Eu apreciaria muito sua ajuda com isso

Answer 1

Isso pode ser feito usando a junção com as colunas 'Q' e 'x' de ambos os conjuntos de dados, contando os valores exclusivos de 'R' e atribuindo-os a uma nova coluna em 'DF1'

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

---

Ou usandotidyverse

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

Answer 2

Uma basesolução com aggregate()e 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