添加依赖于另一列和子集的列(使用tidyverse)

时间:2018-09-18 17:43:15

标签: r dplyr dataset

我有以下股票数据数据集(作为小标题)

Open  Volume   Ticker
<dbl>  <dbl>    <chr>
1106.  10       AAL.L
1086.  30       AAL.L
1043.  9        ABF.L
1055.  2        ABF.L
1048.  90000    BT-A.L
1077.  8000     BT-A.L

使用dplyr程序包,我希望获得此数据集,以及给定股票行情指示器的平均交易量。

Open  Volume   Ticker  AvgVolume
<dbl>  <dbl>    <chr>    <dbl>
1106.  10       AAL.L    20
1086.  30       AAL.L    20
1043.  9        ABF.L    5.5
1055.  2        ABF.L    5.5 
1048.  90000    BT-A.L   49000
1077.  8000     BT-A.L   49000

我将在500000行中执行此操作,因此速度和效率是关键(对于循环等而言则没有……)

我是Tidyverse的新手,希望您能对答案进行解释。

2 个答案:

答案 0 :(得分:2)

尽管您要求一个整洁的解决方案,并且@akrun已经发布了一个解决方案,但我将发布基本的R方式。

这是因为您的问题是适合ave的教科书。

ave(df1$Volume, df1$Ticker)
#[1]    20.0    20.0     5.5     5.5 49000.0 49000.0

只需将此输出分配到新列。

df1$AvgVolume <- ave(df1$Volume, df1$Ticker)
df1
## A tibble: 6 x 4
#   Open Volume Ticker AvgVolume
#  <dbl>  <int> <fct>      <dbl>
#1  1106     10 AAL.L       20  
#2  1086     30 AAL.L       20  
#3  1043      9 ABF.L        5.5
#4  1055      2 ABF.L        5.5
#5  1048  90000 BT-A.L   49000  
#6  1077   8000 BT-A.L   49000

数据。

df1 <-
structure(list(Open = c(1106, 1086, 1043, 1055, 1048, 1077), 
    Volume = c(10L, 30L, 9L, 2L, 90000L, 8000L), Ticker = structure(c(1L, 
    1L, 2L, 2L, 3L, 3L), .Label = c("AAL.L", "ABF.L", "BT-A.L"
    ), class = "factor"), AvgVolume = c(20, 20, 5.5, 5.5, 49000, 
    49000)), row.names = c(NA, -6L), class = c("tbl_df", "tbl", 
"data.frame"))

答案 1 :(得分:1)

我们通过'Ticker'分组,使用mean将'AvgVolume'创建为'Volume'的mutate

library(dplyr)
df1 %>%
   group_by(Ticker) %>%
   mutate(AvgVolume = mean(Volume))
# A tibble: 6 x 4
# Groups:   Ticker [3]
#   Open Volume Ticker AvgVolume
#  <int>  <int> <chr>      <dbl>
#1  1106     10 AAL.L       20  
#2  1086     30 AAL.L       20  
#3  1043      9 ABF.L        5.5
#4  1055      2 ABF.L        5.5
#5  1048  90000 BT-A.L   49000  
#6  1077   8000 BT-A.L   49000