Как удалить строки из кадра данных, если условие?

Question

Я пытаюсь добиться этого с помощью WHILE, но это слишком сложно для меня, должен быть способ использования библиотеки dplyr.

У меня есть склад с:

  product_id amount
1       1001      1
2       4911    100
3       4014     32

IЯ пишу функцию, которая передаст product_id и сумму, и вычтет требуемую сумму, и если такой product_id не существует или сумма, превышающая доступную, возвращает ошибку.

Итак, если я запустил функцию:

remove_warehouse(1001,1)

Результат должен быть:

  product_id amount
1       4911    100
2       4014     32

И если я запусту eiter:

remove_warehouse(240,1)

или

remove_warehouse(4014,60)

Я должен получитьобщая ошибка "недостаточно количества или товара нет"

Answer 1

Одним из способов написания функции может быть

remove_warehouse <- function(df, product_id, amount) {
   id = df$product_id == product_id
   if (any(id))
     amount_base = df$amount[id]
   else
     stop("No id present")
   if (amount > amount_base) 
     stop("No sufficient amount")
   else
     df$amount[id] = df$amount[id] - amount
   df
}

remove_warehouse(df, 4911, 90)

#  product_id amount
#1       1001      1
#2       4911     10
#3       4014     32

remove_warehouse(df, 1234, 12)
#Error in remove_warehouse(df, 1234, 12) : No id present

remove_warehouse(df, 1001, 100)
#Error in remove_warehouse(df, 1001, 100) : No sufficient amount

Это предполагает, что у вас будет только один product_id в ваших df.

данных

df <- structure(list(product_id = c(1001L, 4911L, 4014L), amount = c(1L, 
100L, 32L)), .Names = c("product_id", "amount"), class = "data.frame",
row.names = c("1", "2", "3"))