tidyquant: Bringing Quantitative Financial Analysis to the tidyverse

[This article was first published on mattdancho.com - Articles, and kindly contributed to R-bloggers]. (You can report issue about the content on this page here)
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.

My new package, tidyquant, is now available on CRAN. tidyquant integrates the best quantitative resources for collecting and analyzing quantitative data, xts, quantmod and TTR, with the tidy data infrastructure of the tidyverse allowing for seamless interaction between each. While this post aims to introduce tidyquant to the R community, it just scratches the surface of the features and benefits. We’ll go through a simple stock visualization using ggplot2, which which shows off the integration. The package is open source, and you can view the code on the tidyquant github page.

Table of Contents

Why tidyquant?

One of the reasons why I began my journey into R programming is because it’s the best open-source option for stock analysis. With quantitative financial analysis (QFA) packages like quantmod, xts and TTR, stock data can quickly be retrieved, sliced and diced, transformed and mutated, and visualized so I can make investment decisions. It’s really a beautiful thing.

Over time, the R programming landscape has evolved. A major step forward was the tidyverse, a collection of R packages that work in harmony, are built for scale-ability, and are well documented in R for Data Science. However, a problem has surfaced: the QFA packages are not easy to use with the tidyverse. The tidyverse works with data frames while the QFA packages work with extensible time-series (xts) objects. Both are great, but they don’t easily work together.

As you can imagine, my workflow was longer than I’d like. I’d work in xts to use various functions to calculate moving averages, moving average convergence divergence (MACD), Bollinger Bands, etc, and then convert to tibbles (tidy dataframes) for mapping functions with purrr to scale to many stocks, for mutating dataframes with dplyr to add new columns, and for visualizing my analysis using ggplot2. This got very long and repetitive…

Enter tidyquant. The package started off as a collection of scripts aimed at increasing my efficiency and performance of my stock analyses:

  • I would start by getting data with tq_get(), which returns data, such as stock prices or financial statements, as a tibble object.
  • I’d use tq_transform() to use the various quantmod and xts functions that can change periodicities, such as period returns and conversion from daily to monthly periodicity.
  • I’d use tq_mutate() to seamlessly apply the various TTR functions, such as moving averages, MACD’s, Bollinger Bands, etc.
  • And, I’d do all of this without ever leaving the tidyverse, which allowed me to mutate, pipe (%>%), and scale my analyses at ease.

In this evolution and in the spirit of open source, I have released the tidyquant package to the R community with the hope that others can benefit from the integration between the QFA packages (quantmod, xts, and TTR) and the tidyverse. I believe this is the right way to go, and I’m looking forward to hearing your feedback.

Benefits

The tidyquant philosophy:

  • A few core functions with a lot of power, that
  • leverage the quantitative analysis power of xts, quantmod and TTR, and are
  • designed to be used and scaled with the tidyverse.

Example: Visualizing Moving Averages

I’ll go through an example of visualizing the 15-day and 50-day moving averages of the stock symbol, AAPL, which is for Apple Inc. Moving averages are a popular trading tool that stock analysts use to determine buying and selling signals. According to Investopedia, the moving average is…

A widely used indicator in technical analysis that helps smooth out price action by filtering out the “noise” from random price fluctuations. A moving average (MA) is a trend-following or lagging indicator because it is based on past prices. The two basic and commonly used MAs are the simple moving average (SMA), which is the simple average of a security over a defined number of time periods, and the exponential moving average (EMA), which gives bigger weight to more recent prices. The most common applications of MAs are to identify the trend direction and to determine support and resistance levels. While MAs are useful enough on their own, they also form the basis for other indicators such as the Moving Average Convergence Divergence (MACD).

Of particular interest is the crossover, the point at which a trend begins to emerge, which can be used as a buy or sell signal.

Moving Averages

Source: Investopedia: Moving Averages

Let’s go through an example to visualize the 15-day and 50-day moving averages for AAPL.

Step 1: Prerequisites

The tidyquant package can be downloaded from CRAN:

install.packages("tidyquant")

For those following along in R, you’ll need to load the following package:

# Loads tidyquant, tidyverse, lubridate, quantmod, TTR, and xts
library(tidyquant)

I also recommend the open-source RStudio IDE, which makes R Programming easy and efficient.

Step 2: Use tq_get to get stock prices

We’ll start by getting the last year of stock prices. We use the tidyquant tq_get() function for all data retrieval. Set the parameter get = "stock.prices" to tell tidyquant we want the historical stock prices. We can use the from argument to pass a date as the start of the collection, which accepts character string in the form of “YYYY-MM-DD”. We can use lubridate functions today() and years() to get the date from one year ago.

from <- today() - years(1)
AAPL <- tq_get("AAPL", get = "stock.prices", from = from)
AAPL
## # A tibble: 252 × 7
##          date   open   high    low  close   volume  adjusted
##        <date>  <dbl>  <dbl>  <dbl>  <dbl>    <dbl>     <dbl>
## 1  2016-01-04 102.61 105.37 102.00 105.35 67649400 103.05706
## 2  2016-01-05 105.75 105.85 102.41 102.71 55791000 100.47452
## 3  2016-01-06 100.56 102.37  99.87 100.70 68457400  98.50827
## 4  2016-01-07  98.68 100.13  96.43  96.45 81094400  94.35077
## 5  2016-01-08  98.55  99.11  96.76  96.96 70798000  94.84967
## 6  2016-01-11  98.97  99.06  97.34  98.53 49739400  96.38550
## 7  2016-01-12 100.55 100.69  98.84  99.96 49154200  97.78438
## 8  2016-01-13 100.32 101.19  97.30  97.39 62439600  95.27031
## 9  2016-01-14  97.96 100.48  95.74  99.52 63170100  97.35395
## 10 2016-01-15  96.20  97.71  95.36  97.13 79010000  95.01597
## # ... with 242 more rows

We now have 252 days of stock prices as a tibble object. This is exactly the format we want for working in the tidyverse.

Step 3: Use tq_mutate to add moving averages

We need to get the 15-day and 50-day moving averages. We want to use the SMA() function from the TTR package. To use any of these functions in the tidyverse, we have a few options with pros and cons:

  • dplyr::mutate(): Used to add a single column to a data set. Only able to add a single column to a tibble. For SMA(), this works because a single column is generated. For other functions such as BBands() and MACD, multiple columns are generated that fail on mutate().

  • tidyquant::tq_mutate(): Used to add single or multiple columns to a data set. Uses quantmod OHLC notation (more on this in a minute). The output generated must be the number of rows as the input dataframe (otherwise the data can’t be joined). Because multiple columns can be returned, works with BBands() and MACD().

  • tidyquant::tq_transform(): Used to return a new data set with output only (does not return the input dataframe). Uses OHLC notation. Most flexible option.

  • tidyquant::tq_mutate_xy(): Same as tq_mutate() but works using up to two column inputs instead of OHLC notation.

  • tidyquant::tq_transform_xy(): Same as tq_transform() but works using up to two column inputs instead of OHLC notation.

For this tutorial, we will use tq_mutate() to expose you to OHLC notation along with the tidyquant function workflow. We’ll also show tq_mutate_xy() so you can see the difference in arguments.

tq_mutate() function

tq_mutate() has two primary arguments: x_fun and mutate_fun:

  • x_fun: Takes quantmod::OHLC functions, which are Op, Cl, Hi, Lo, Vo, Ad, HLC, OHLC, and OHLCV. The OHLC notation is the basis of all quantmod, xts, and TTR functions. These functions collect a subset of the dataframe columns matching open, high, low, close, volume, and/or adjusted. Think of the OHLC notation akin to the dplyr::select() function, which selects columns. Op selects the column named “open”, and HLC selects “high”, “low” and “close” columns.

  • mutate_fun: Takes any quantmod, xts, or TTR function listed in tq_mutate_fun_options() (see below for compatible functions). The mutation_fun performs the work. Any additional parameters of the passed via ... in the tq_mutate() function go to the mutation_fun.

# Get list of tq_mutate() options
tq_mutate_fun_options() %>% 
    str() # str() to minimize output. Remove str() to see full output.
## List of 3
##  $ xts     : chr [1:27] "apply.daily" "apply.monthly" "apply.quarterly" "apply.weekly" ...
##  $ quantmod: chr [1:25] "allReturns" "annualReturn" "ClCl" "dailyReturn" ...
##  $ TTR     : chr [1:61] "adjRatios" "ADX" "ALMA" "aroon" ...

An example with SMA() from the TTR package helps solidify how it works. Reviewing the documentation for SMA, we see that the function, SMA(x, n = 10, ...), accepts x a price or volume and n a number of periods to average over. For the 15-day simple moving average, we would pass a set of prices, either “close” or “adjusted”, and n = 15 for 15 days. In OHLC notation x_fun = Cl for “close” or x_fun = Ad for adjusted. The mutate_fun = SMA, and we pass n = 15 as an additional argument. Shown below, we pipe (%>%) our tibble of AAPL stock prices to tq_mutate(x_fun = Cl, mutate_fun = SMA, n = 15), which creates an additional column with the simple moving average of the close prices.

AAPL %>%
    tq_mutate(x_fun = Cl, mutate_fun = SMA, n = 15)
## # A tibble: 252 × 8
##          date   open   high    low  close   volume  adjusted   SMA
##        <date>  <dbl>  <dbl>  <dbl>  <dbl>    <dbl>     <dbl> <dbl>
## 1  2016-01-04 102.61 105.37 102.00 105.35 67649400 103.05706    NA
## 2  2016-01-05 105.75 105.85 102.41 102.71 55791000 100.47452    NA
## 3  2016-01-06 100.56 102.37  99.87 100.70 68457400  98.50827    NA
## 4  2016-01-07  98.68 100.13  96.43  96.45 81094400  94.35077    NA
## 5  2016-01-08  98.55  99.11  96.76  96.96 70798000  94.84967    NA
## 6  2016-01-11  98.97  99.06  97.34  98.53 49739400  96.38550    NA
## 7  2016-01-12 100.55 100.69  98.84  99.96 49154200  97.78438    NA
## 8  2016-01-13 100.32 101.19  97.30  97.39 62439600  95.27031    NA
## 9  2016-01-14  97.96 100.48  95.74  99.52 63170100  97.35395    NA
## 10 2016-01-15  96.20  97.71  95.36  97.13 79010000  95.01597    NA
## # ... with 242 more rows

We need both the 15-day and the 50-day moving average, which is two steps with the pipe. I rename in between steps so the column names are more descriptive.

AAPL %>%
    tq_mutate(x_fun = Cl, mutate_fun = SMA, n = 15) %>%
    rename(SMA.15 = SMA) %>%
    tq_mutate(x_fun = Cl, mutate_fun = SMA, n = 50) %>%
    rename(SMA.50 = SMA)
## # A tibble: 252 × 9
##          date   open   high    low  close   volume  adjusted SMA.15
##        <date>  <dbl>  <dbl>  <dbl>  <dbl>    <dbl>     <dbl>  <dbl>
## 1  2016-01-04 102.61 105.37 102.00 105.35 67649400 103.05706     NA
## 2  2016-01-05 105.75 105.85 102.41 102.71 55791000 100.47452     NA
## 3  2016-01-06 100.56 102.37  99.87 100.70 68457400  98.50827     NA
## 4  2016-01-07  98.68 100.13  96.43  96.45 81094400  94.35077     NA
## 5  2016-01-08  98.55  99.11  96.76  96.96 70798000  94.84967     NA
## 6  2016-01-11  98.97  99.06  97.34  98.53 49739400  96.38550     NA
## 7  2016-01-12 100.55 100.69  98.84  99.96 49154200  97.78438     NA
## 8  2016-01-13 100.32 101.19  97.30  97.39 62439600  95.27031     NA
## 9  2016-01-14  97.96 100.48  95.74  99.52 63170100  97.35395     NA
## 10 2016-01-15  96.20  97.71  95.36  97.13 79010000  95.01597     NA
## # ... with 242 more rows, and 1 more variables: SMA.50 <dbl>

tq_mutate_xy() function

Not all quantmod, xts, and TTR functions work with OHLC notation. A few of these functions take two primary inputs. An example of this is the Delt function from the quantmod package. The function form is Delt(x1, x2 = NULL, k = 0, type = c("arithmetic", "log")), which has x1 and x2 arguments. In these situations you will need to use the XY variant, tq_mutate_xy(), which accepts .x (required) and .y (optional). For the Delt function, .x = x1 and .y = x2.

For the SMA() function, we don’t need the .y argument, but we can use the XY variant to accomplish the same task as the OHLC variant. The operation is the same except instead of x_fun = Cl we replace with .x = close (the name of the column being passed to the mutation function).

AAPL %>%
    tq_mutate_xy(.x = close, mutate_fun = SMA, n = 15) %>%
    rename(SMA.15 = SMA) %>%
    tq_mutate_xy(.x = close, mutate_fun = SMA, n = 50) %>%
    rename(SMA.50 = SMA)
## # A tibble: 252 × 9
##          date   open   high    low  close   volume  adjusted SMA.15
##        <date>  <dbl>  <dbl>  <dbl>  <dbl>    <dbl>     <dbl>  <dbl>
## 1  2016-01-04 102.61 105.37 102.00 105.35 67649400 103.05706     NA
## 2  2016-01-05 105.75 105.85 102.41 102.71 55791000 100.47452     NA
## 3  2016-01-06 100.56 102.37  99.87 100.70 68457400  98.50827     NA
## 4  2016-01-07  98.68 100.13  96.43  96.45 81094400  94.35077     NA
## 5  2016-01-08  98.55  99.11  96.76  96.96 70798000  94.84967     NA
## 6  2016-01-11  98.97  99.06  97.34  98.53 49739400  96.38550     NA
## 7  2016-01-12 100.55 100.69  98.84  99.96 49154200  97.78438     NA
## 8  2016-01-13 100.32 101.19  97.30  97.39 62439600  95.27031     NA
## 9  2016-01-14  97.96 100.48  95.74  99.52 63170100  97.35395     NA
## 10 2016-01-15  96.20  97.71  95.36  97.13 79010000  95.01597     NA
## # ... with 242 more rows, and 1 more variables: SMA.50 <dbl>

Back to the example

Returning back to our need, we get the simple moving averages using one of the the code options mentioned previously.

AAPL <- AAPL %>%
    tq_mutate(x_fun = Cl, mutate_fun = SMA, n = 15) %>%
    rename(SMA.15 = SMA) %>%
    tq_mutate(x_fun = Cl, mutate_fun = SMA, n = 50) %>%
    rename(SMA.50 = SMA)

Step 4: Visualize the Simple Moving Averages

We have our 15-day and 50-day simple moving averages. Now all we need to do is visualize using ggplot2. The format of the data will need to be tidy, which requires us to use gather() from the tidyr package to shift the close, SMA.15, and SMA.50 columns into a long form with type and price. The code and final data form is shown below.

AAPL %>%
    select(date, close, SMA.15, SMA.50) %>%
    gather(key = type, value = price, close:SMA.50)
## # A tibble: 756 × 3
##          date  type  price
##        <date> <chr>  <dbl>
## 1  2016-01-04 close 105.35
## 2  2016-01-05 close 102.71
## 3  2016-01-06 close 100.70
## 4  2016-01-07 close  96.45
## 5  2016-01-08 close  96.96
## 6  2016-01-11 close  98.53
## 7  2016-01-12 close  99.96
## 8  2016-01-13 close  97.39
## 9  2016-01-14 close  99.52
## 10 2016-01-15 close  97.13
## # ... with 746 more rows

Now, we can use ggplot2 to plot the tidy data. We use the same select and gather statements above and pipe to ggplot. I add a custom palette to match the black, blue and red colors from the Investopedia graphic. The final code chunk for the visualization is as follows:

my_palette <- c("black", "blue", "red")
AAPL %>%
    select(date, close, SMA.15, SMA.50) %>%
    gather(key = type, value = price, close:SMA.50) %>%
    ggplot(aes(x = date, y = price, col = type)) +
    geom_line() +
    scale_colour_manual(values = my_palette) + 
    theme(legend.position="bottom") +
    ggtitle("Simple Moving Averages are a Breeze with tidyquant") +
    xlab("") + 
    ylab("Stock Price")

plot of chunk unnamed-chunk-10

Conclusion

The tidyquant package integrates the three primary QFA packages, quantmod, xts, and TTR, with the tidyverse.

Recap

The purpose of this post was twofold:

  1. Introduce you to the tidyquant package
  2. Show an example of the integration between the QFA packages and the tidyverse.

We discussed why there is a need for tidyquant, which is to help minimize the back and forth between xts and tibble (tidy dataframes). We also went through an example of getting simple moving averages, which previously required jumping back and forth between xts and tibble objects. The tidyquant package made this much easier.

This example just scratches the surface of the power of tidyquant. See the vignette for a detailed discussion on each of the tidyquant features.

Further Reading

  1. tidyquant Vignette: This tutorial just scratches the surface of tidyquant. The vignette explains much, much more!

  2. R for Data Science: A free book that thoroughly covers the tidyverse packages.

  3. Quantmod Website: Covers many of the quantmod functions. Also, see the quantmod vignette.

  4. Extensible Time-Series Website: Covers many of the xts functions. Also, see the xts vignette.

  5. TTR Vignette: Covers each of the TTR functions.

To leave a comment for the author, please follow the link and comment on their blog: mattdancho.com - Articles.

R-bloggers.com offers daily e-mail updates about R news and tutorials about learning R and many other topics. Click here if you're looking to post or find an R/data-science job.
Want to share your content on R-bloggers? click here if you have a blog, or here if you don't.

Never miss an update!
Subscribe to R-bloggers to receive
e-mails with the latest R posts.
(You will not see this message again.)

Click here to close (This popup will not appear again)