Question

我有一个项目编号和库存位置列表。商品可以显示在两种类型的广告资源位置 - warehouse或par位置。我想在我的数据中添加一列，以指示特定项目编号是否与仓库位置相关联。示例数据如下：

item_num <- c("Item - 1", "Item - 2", "Item - 3", "Item - 1", "Item - 3", 
          "Item - 2")
locs <- c("warehouse", "par", "par", "par", "warehouse", "par")
fake_data <- tibble(item_num, locs)

fake_data
# A tibble: 6 x 2
  item_num      locs
     <chr>     <chr>
1 Item - 1 warehouse
2 Item - 2       par
3 Item - 3       par
4 Item - 1       par
5 Item - 3 warehouse
6 Item - 2       par

我想为所有TRUE和Item - 1添加一个Item - 3列，因为我的数据中有两个仓库位置。对于Item - 2，这将是错误的。像这样：

  item_num      locs wh_exists
1 Item - 1 warehouse      TRUE
2 Item - 2       par     FALSE
3 Item - 3       par      TRUE
4 Item - 1       par      TRUE
5 Item - 3 warehouse      TRUE
6 Item - 2       par     FALSE

我的凌乱解决方案如下。

wh_locs <- fake_data %>%
  filter(locs == "warehouse")
fake_data$wh_exist <- fake_data$item_num %in% wh_locs$item_num

fake_data

# A tibble: 6 x 3
  item_num      locs wh_exist
     <chr>     <chr>    <lgl>
1 Item - 1 warehouse     TRUE
2 Item - 2       par    FALSE
3 Item - 3       par     TRUE
4 Item - 1       par     TRUE
5 Item - 3 warehouse     TRUE
6 Item - 2       par    FALSE

这很有效，但在我看来，应该有一些聪明的方法来使用mutate和group_by来获得答案，这样我就可以将它全部保存在一组管道功能中。

谢谢。

Answer 1

我们可以将any与==

一起使用

fake_data %>%
      group_by(item_num) %>% 
      mutate(wh_exists = any(locs == "warehouse"))
#     item_num      locs wh_exists
#     <chr>     <chr>     <lgl>
#1 Item - 1 warehouse      TRUE
#2 Item - 2       par     FALSE
#3 Item - 3       par      TRUE
#4 Item - 1       par      TRUE
#5 Item - 3 warehouse      TRUE
#6 Item - 2       par     FALSE

使用data.table

的类似选项

library(data.table)
setDT(fake_data)[, wh_exists := any(locs == "warehouse"), by = item_num]

Answer 2

将group_by与%in%运算符一起使用，检查warehouse是否在每个群组的locs范围内：

library(dplyr)
fake_data %>% group_by(item_num) %>% mutate(wh_exists = 'warehouse' %in% locs)

# Source: local data frame [6 x 3]
# Groups: item_num [3]
# 
#   item_num      locs wh_exists
#     <fctr>    <fctr>     <lgl>
# 1 Item - 1 warehouse      TRUE
# 2 Item - 2       par     FALSE
# 3 Item - 3       par      TRUE
# 4 Item - 1       par      TRUE
# 5 Item - 3 warehouse      TRUE
# 6 Item - 2       par     FALSE

R dplyr - 根据其他行的结果添加列

2 个答案: