如何根据R中的条件替换列值

时间:2019-11-24 22:21:18

标签: r data-extraction

我有一个不同订单的数据集以及该订单给客户的数量。 我想从所有行中删除ordertype ==“ cap”,并为该订单的相应数量形成相应的数量列,然后将其替换为与“ cap”不对应的下一个值

#INPUT DATA 

custID <- data.frame(c(1,2,3,4,5))
OrderType_1 <- data.frame(c("ball", "pen", "ball", "shuttle", "pen"))
OrderType_2 <- data.frame(c("pen", NA, "cap", "cap", "pen"))
OrderType_3 <- data.frame(c("cap", NA, "cap", "cap", NA))
OrderType_4 <- data.frame(c("shuttle", NA, "ball", "cap", NA))
OrderType_5 <- data.frame(c("pen", NA, "cap", "ball", NA))
QUANTITY_1 <- data.frame(c(2,3,4,5,6))
QUANTITY_2 <- data.frame(c(2, NA, 1, 3, 3))
QUANTITY_3 <- data.frame(c(3,NA,5,6,NA))
QUANTITY_4 <- data.frame(c(2,NA,3,5,NA))
QUANTITY_5 <- data.frame(c(2,NA,2,3, NA))

report <- cbind(custID, OrderType_1, OrderType_2, OrderType_3, OrderType_4, 
OrderType_5, QUANTITY_1, QUANTITY_2, QUANTITY_3, QUANTITY_4, QUANTITY_5 )
report <- as.data.frame(report)

colnames(report) <- c("CustID", "OrderType_1", "OrderType_2", "OrderType_3", 
"OrderType_4", "OrderType_5", "QUANTITY_1", "QUANTITY_2", "QUANTITY_3", 
"QUANTITY_4", "QUNATITY_5")

这是除去“上限”和相应数量值后输出的外观。

#OUTPUT DATA TYPE

custID <- data.frame(c(1,2,3,4,5))
OrderType_1 <- data.frame(c("ball", "pen", "ball", "shuttle", "pen"))
OrderType_2 <- data.frame(c("pen", NA, "ball", "ball", "pen"))
OrderType_3 <- data.frame(c("shuttle", NA, NA, NA, NA))
OrderType_4 <- data.frame(c("pen", NA, NA, NA, NA))
OrderType_5 <- data.frame(c(NA, NA, NA, NA, NA))
QUANTITY_1 <- data.frame(c(2,3,4,5,6))
QUANTITY_2 <- data.frame(c(2, NA, 3, 3, 3))
QUANTITY_3 <- data.frame(c(2,NA,NA,NA,NA))
QUANTITY_4 <- data.frame(c(2, NA,NA,5,NA))
QUANTITY_5 <- data.frame(c(NA,NA,NA,NA,NA))

report_1 <- cbind(custID, OrderType_1, OrderType_2, OrderType_3, 
OrderType_4, OrderType_5, QUANTITY_1, QUANTITY_2, QUANTITY_3, QUANTITY_4, 
QUANTITY_5 )
report_1 <- as.data.frame(report_1)

colnames(report_1) <- c("CustID", "OrderType_1", "OrderType_2", 
"OrderType_3", 
"OrderType_4", "OrderType_5", "QUANTITY_1", "QUANTITY_2", "QUANTITY_3", 
"QUANTITY_4", "QUNATITY_5")

1 个答案:

答案 0 :(得分:2)

也许可以使用tidyverse来做到这一点:

使用pivot_longer可以更轻松地操纵这些数据。您可以过滤掉不需要的行(同时删除OrderTypeQUANTITY)。然后,如果需要的格式为pivot_wider,则根据需要填写NA。我希望这会有所帮助。

修改:在过滤掉不需要的订单后,我需要为每个CustID重新排序。

library(tidyverse)

report %>%
  pivot_longer(cols = -CustID, 
               names_to = c(".value", "order"),
               names_sep = "_") %>%
  filter(OrderType != "cap") %>%
  group_by(CustID) %>%
  mutate(neworder = row_number()) %>%
  pivot_wider(id_cols = CustID, 
              names_from = c(neworder, neworder), 
              names_sep = "_", 
              values_from = c(OrderType, QUANTITY))

# A tibble: 5 x 9
# Groups:   CustID [5]
  CustID OrderType_1 OrderType_2 OrderType_3 OrderType_4 QUANTITY_1 QUANTITY_2 QUANTITY_3 QUANTITY_4
   <dbl> <fct>       <fct>       <fct>       <fct>            <dbl>      <dbl>      <dbl>      <dbl>
1      1 ball        pen         shuttle     pen                  2          2          2          2
2      2 pen         NA          NA          NA                   3         NA         NA         NA
3      3 ball        ball        NA          NA                   4          3         NA         NA
4      4 shuttle     ball        NA          NA                   5          3         NA         NA
5      5 pen         pen         NA          NA                   6          3         NA         NA