根据涉及其他列的公式创建新列

Question

首先，我想根据列“类型”(1 : 5) 定义的产品类型创建一个名为“总价”的新列。数据如下所示：

df <- data.frame('type'=c(1,2,3,4,5), 'quantity'=c(1,20,3,4,50))
df
  type quantity
1   1    1         
2   2   20
3   3    3
4   4    4
5   5   50

总价可以通过每个产品的数量乘以相应的价格来计算：

***每种产品的价格：Type 1 - $10； 2 型 - 20 美元；3 型 - 30 美元；4 型 - 40 美元；5 型 - 50 美元

所以我希望看起来像这样：

       type quantity totalprice
    1   1    1           10
    2   2   20          400
    3   3    3           90
    4   4    4          160
    5   5   50         2500

然后，我需要将这个新创建的列与原始数据作为最后一列加入。

我猜伪代码看起来像这样：

Part 1)
if df$type == "1" , totalprice = df$quantity * 10,
if df$type == "2", totalprice = df$quantity * 20,
if df$type == "3", totalprice = df$quantity * 30,
if df$type == "4", totalprice = df$quantity * 40

Part2)
Collect these values in a column called totalprice and join it to df.

非常感谢您的帮助。

Answer 1

我同意@Elin，创建一个包含所有类型及其对应price 的参考数据框。 merge 将它与你原来的 df 乘以 "type" 然后乘以 quantity * price

ref_data <- data.frame(type = 1:5, price = c(10, 20, 30, 40, 50))
transform(merge(df, ref_data, by = "type"), totalprice = quantity * price)

#  type quantity price totalprice
#1    1        1    10         10
#2    2       20    20        400
#3    3        3    30         90
#4    4        4    40        160
#5    5       50    50       2500

在dplyr，你可以这样做

library(dplyr)
inner_join(df, ref_data, by = "type") %>%
    mutate(totalprice = quantity * price)

Answer 2

我建议只添加一个与type 对应的乘数列，然后使用它来生成总价：

df$multiplier <- 10
df$totalprice <- df$multiplier * df$quantity

Answer 3

如果对应的价格有严格的规则（类型X 10），可以通过这段代码生成：

require(dplyr)
df %>%
  mutate(price = type * 10,
         totalprice = quantity * price)

df 有更多行吗？还是没有价格规则？