基本上,对于每个ID,我都有一组产品ID,并且我尝试将它们分布在一组定义列中。每个ID只能有5个product_id。 例如:
id product_id
1 305
1 402
2 200
1 305
3 402
3 402
所以我以二进制结果的形式传播:
id 305 402 200
1 2 0 0
2 0 0 1
3 0 2 0
但是我想要:
id product1 product2 product3 product4... until 5
1 305 305 0
2 200 0 0
3 402 402 0
如果有人有干净的东西(我大约有1万行),那太棒了!谢谢!
#this gives me the binary outcome
for (i in names(test2[2:18])) {
test2$product1[test2[i] == 1 ] <- i
}
#this is a try to iterate through each row but it s pretty bad
for(i in 1:nrow(test2)){
if(test2[i,1]== 1){
test2$product1[i] <- colnames(test2[1])
} else if(test2[i,1]==2){
test2$product1[i] <- colnames(test2[1])
test2$product2[i] <- colnames(test2[1])
} else if(test2[i,1]==3){
test2$product1[i] <- colnames(test2[1])
test2$product2[i] <- colnames(test2[1])
test2$product3[i] <- colnames(test2[1])
} else if(test2[i,1]==4){
and so one...
预期:
id product1 product2 product3 product4... until 5
1 305 305 0
2 200 0 0
3 402 402 0
实际:
id 305 402 200
1 2 0 0
2 0 0 1
3 0 2 0
答案 0 :(得分:1)
我们可以先按'id'然后按spread
创建一个序列列。请注意,仅spread
ing不会拥有所有的“乘积”直到5,因为数据中缺少这些乘积。为此,将序列创建为factor
,其中levels
从'product1'到'product5',并在spread
中指定drop = FALSE
,以免删除未使用的levels
library(tidyverse)
df1 %>%
group_by(id) %>%
mutate(product = factor(paste0('product', row_number()),
levels = paste0('product', 1:5))) %>%
spread(product, product_id, drop = FALSE, fill = 0)
# A tibble: 3 x 6
# Groups: id [3]
# id product1 product2 product3 product4 product5
# <int> <dbl> <dbl> <dbl> <dbl> <dbl>
#1 1 305 402 305 0 0
#2 2 200 0 0 0 0
#3 3 402 402 0 0 0
df1 <- structure(list(id = c(1L, 1L, 2L, 1L, 3L, 3L), product_id = c(305L,
402L, 200L, 305L, 402L, 402L)), class = "data.frame", row.names = c(NA,
-6L))