循环以跨列相乘

时间:2017-11-02 03:23:26

标签: r dataframe tidyverse mutate

我的数据框中标有sales1sales2price1price2的列,我希望通过乘以sales1 * {{来计算收入1}}等等以迭代的方式跨越每个数字。

price1

为什么以下代码不起作用?

data <- data_frame(
  "sales1" = c(1, 2, 3),
  "sales2" = c(2, 3, 4),
  "price1" = c(3, 2, 2),
  "price2" = c(3, 3, 5))

data
# A tibble: 3 x 4
#  sales1 sales2 price1 price2
#   <dbl>  <dbl>  <dbl>  <dbl>
#1      1      2      3      3
#2      2      3      2      3
#3      3      4      2      5

2 个答案:

答案 0 :(得分:3)

假设您的列已经订购(sales1sales2price1price2)。我们可以将数据帧分成两部分,然后将它们相乘

data[grep("sales", names(data))] * data[grep("price", names(data))]

#  sales1 sales2
#1      3      6
#2      4      9
#3      6     20

如果列尚未按名称排序,我们可以使用order对其进行排序,然后使用上面的命令。

data <- data[order(names(data))]

答案 1 :(得分:1)

这个答案并不简短。为此,@ RonakShah现有的答案是值得关注的!

我的回答旨在解决有关在tidyverse尝试执行此操作的难度的更广泛关注。我的理解是这很困难,因为数据目前还没有整理好。格式。相反,您可以创建一个整洁的数据框,如下所示:

library(tidyverse)

tidy_df <- data %>% 
  rownames_to_column() %>%
  gather(key, value, -rowname) %>% 
  extract(key, c("variable", "id"), "([a-z]+)([0-9]+)") %>%
  spread(variable, value)

然后使最终计算变得简单

tidy_df %>% mutate(revenue = sales * price)

#> # A tibble: 6 x 5
#>   rowname    id price sales revenue
#>     <chr> <chr> <dbl> <dbl>   <dbl>
#> 1       1     1     3     1       3
#> 2       1     2     3     2       6
#> 3       2     1     2     2       4
#> 4       2     2     3     3       9
#> 5       3     1     2     3       6
#> 6       3     2     5     4      20

如果您需要将数据恢复为原始格式,您可以对我感到笨拙(我确信这可以在某些方面得到改善)。

tidy_df %>% mutate(revenue = sales * price) %>%
  gather(key, value, -c(rowname, id)) %>%
  unite(key, key, id, sep = "") %>%
  spread(key, value) %>% 
  select(starts_with("price"), 
         starts_with("sales"),
         starts_with("revenue"))

#> # A tibble: 3 x 6
#>   price1 price2 sales1 sales2 revenue1 revenue2
#> *  <dbl>  <dbl>  <dbl>  <dbl>    <dbl>    <dbl>
#> 1      3      3      1      2        3        6
#> 2      2      3      2      3        4        9
#> 3      2      5      3      4        6       20