Jak utworzyć zmienną na podstawie liczby unikalnych wartości w innej ramce danych?

Aug 16 2020

To jest uproszczony przykład tego, co chcę zrobić.

Zestaw danych 1 (DF1) zawiera dane jabłek (takie jak rozmiar lub liczba otworów), a drugi zestaw danych (DF2) zawiera informacje o znalezionych w nich robakach, w tym o kolorze i jabłku, w którym zostały znalezione. Chcę dodać zmienną do DF1 z liczbą unikalnych kolorów (robaków), które istnieją w każdym jabłku.

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

Jestem nowy w R i próbując go rozwiązać pomyślałem:

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

Ale zwraca mi nową zmienną wypełnioną zerami zamiast poszukiwanego wektora:

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

Byłbym bardzo wdzięczny za twoją pomoc

Odpowiedzi

3 akrun Aug 16 2020 at 02:11

Można to zrobić, wykorzystując sprzężenie z kolumnami „Q”, „x” obu zbiorów danych, policzyć unikalne wartości „R” i przypisać je do nowej kolumny w „DF1”

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

Lub używając tidyverse

library(dplyr)
DF2 %>%
   group_by(x = Q) %>%
   summarise(N.Colors = n_distinct(R)) %>%
   right_join(DF1)
3 DarrenTsai Aug 16 2020 at 02:19

baseRoztwór o aggregate()i 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