gpt4 book ai didi

r - 使用 Tidyr/Dplyr 汇总字符串组的计数

转载 作者:行者123 更新时间:2023-12-03 21:09:02 26 4
gpt4 key购买 nike

我需要总结我分配给组的字符串计数,我知道我可以在 dplyr/tidyr 中完成,但我遗漏了一些东西。

示例数据集:

Owner = c('bob','julia','cheryl','bob','julia','cheryl')
Day = c('Mon', 'Tue')
Locn = c('house','store','apartment','office','house','shop')
data <- data.frame(Owner, Day, Locn)

看起来像这样:
   Owner Day      Locn
1 bob Mon house
2 julia Tue store
3 cheryl Mon apartment
4 bob Tue office
5 julia Mon house
6 cheryl Tue shop

我想按名称和日期分组,然后在列中计算分组位置。在此示例中,我希望将“房子”和“公寓”添加到标题为“家”的列中,并将“商店”、“办公室”和“商店”添加到“工作”列中。

我当前的代码(不起作用):
grouped_locn <- data %>%
dplyr::arrange(Owner, Day) %>%
dplyr::group_by(Owner, Day) %>%
dplyr::summarize(Home = which(data$Locn %in% c('house', 'apartment')),
Work = which(data$Locn %in% c("store", "office", "apartment")))

我只包括了我目前在总结步骤中的尝试,以展示我是如何接近它的。 Home 和 Work 代码当前返回包含组元素的行号的向量(即 Home = 1 3 5)

我的预期输出:
   Owner Day   Home  Work
1 bob Mon 1 0
2 bob Tue 0 1
3 julia Mon 1 0
4 julia Tue 0 1
5 cheryl Mon 1 0
6 cheryl Tue 0 1

在实际数据集(30k+ 行)中,每个所有者每天有多个 Locn 值,因此家庭和工作计数可以是 1 和 0 以外的数字(因此没有 bool 值)。

非常感谢。

最佳答案

这是一个使用 data.table 的简单有效的解决方案

对于旧版本 (v < 1.9.5)

library(data.table) # v < 1.9.5
setDT(data)[, Locn2 := c("Work", "Home")[(Locn %in% c('house', 'apartment')) + 1L]]
dcast.data.table(data, Owner + Day ~ Locn2, length)
# Owner Day Home Work
# 1: bob Mon 1 0
# 2: bob Tue 0 1
# 3: cheryl Mon 1 0
# 4: cheryl Tue 0 1
# 5: julia Mon 1 0
# 6: julia Tue 0 1

对于较新的版本 (v >= 1.9.5),您可以在一行中执行此操作
dcast(setDT(data), Owner + Day ~ c("Work", "Home")[(Locn %in% c('house', 'apartment')) + 1L], length)

这是一个 tidyr选择
library(dplyr)
library(tidyr)
data %>%
mutate(temp = 1L,
Locn = ifelse(Locn %in% c('house', 'apartment'), "Home", "Work")) %>%
spread(Locn, temp, fill = 0L)

# Owner Day Home Work
# 1 bob Mon 1 0
# 2 bob Tue 0 1
# 3 cheryl Mon 1 0
# 4 cheryl Tue 0 1
# 5 julia Mon 1 0
# 6 julia Tue 0 1

关于r - 使用 Tidyr/Dplyr 汇总字符串组的计数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31453203/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com