Mam te dwie tabele;
<A> <B>
a1 a2 b1
ABC CAFE AB
ABD DRINK BF
ABF CAFE ..
ABFF DRINK
.. ..
Chciałbym poznać tabelę podsumowującą zawierającą B do a1 w tabeli A w ten sposób;
library(dplyr)
library(stringr)
A1 <- A %>%
filter(str_detect(a1, "AB")) %>%
group_by(a2) %>%
summarize(n())
A2 <- A %>%
filter(str_detect(a1, "BF")) %>%
group_by(a2) %>%
summarize(n())
Powinienem jednak wykonać kod kilka razy, aby funkcja mogła wprowadzić tabelę B w funkcji str_detect ... Jak wykonać tę funkcję?
Odpowiedzi:
1 dla odpowiedzi № 1Tutaj zaprojektowałem funkcję o nazwie count_fun
, który ma cztery argumenty. dat
jest jak ramka danych A
, Scol
to kolumna z ciągami znaków, Gcol
to kolumna grupująca, oraz String
to ciąg testowy. Widzieć https://cran.r-project.org/web/packages/dplyr/vignettes/programming.html aby dowiedzieć się, jak zaprojektować funkcję za pomocą dplyr
.
library(dplyr)
library(stringr)
count_fun <- function(dat, Scol, Gcol, String){
Scol <- enquo(Scol)
Gcol <- enquo(Gcol)
dat2 <- dat %>%
filter(str_detect(!!Scol, String)) %>%
group_by(!!Gcol) %>%
summarize(n())
return(dat2)
}
count_fun(A, a1, a2, "AB")
# # A tibble: 2 x 2
# a2 `n()`
# <chr> <int>
# 1 CAFE 2
# 2 DRINK 2
count_fun(A, a1, a2, "BF")
# # A tibble: 2 x 2
# a2 `n()`
# <chr> <int>
# 1 CAFE 1
# 2 DRINK 1
Możemy wtedy złożyć wniosek count_fun
za pomocą lapply
aby zapętlić wszystkie elementy B
.
lapply(B$b1, function(x){
count_fun(A, a1, a2, x)
})
# [[1]]
# # A tibble: 2 x 2
# a2 `n()`
# <chr> <int>
# 1 CAFE 2
# 2 DRINK 2
#
# [[2]]
# # A tibble: 2 x 2
# a2 `n()`
# <chr> <int>
# 1 CAFE 1
# 2 DRINK 1
DANE
A <- read.table(text = "a1 a2
ABC CAFE
ABD DRINK
ABF CAFE
ABFF DRINK
",
header = TRUE, stringsAsFactors = FALSE)
B <- data.frame(b1 = c("AB", "BF"), stringsAsFactors = FALSE)
1 dla odpowiedzi nr 2
Myślę, że to rozwiązało twój problem:
lapply(B$b1,function(x)A%>%filter(str_detect(a1, x)) %>% group_by(a2) %>% summarize(n()))