定义列中的spread()

时间:2018-12-19 21:08:15

标签: r

基本上,对于每个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

1 个答案:

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