Python Itertools Tutorial – Part II

Articles From: QuantInsti
Website: QuantInsti

By:

QuantInsti

See Part I in this series to get familiar with itertools.

Terminating iterators

In contrast to the infinite iterators, this type of iterator does not keep going endlessly. Terminating iterators produce a short output and are used for fast processing of the elements in a collection. Let’s go through a few of them now.

The accumulate() iterator

This iterator can be used to perform algebraic operations on the elements of a collection. For example, let’s say we have the daily percentage returns of the closing price of Tesla, Inc. (TSLA) and we want to see how it adds up. Well we will use the accumulate function.

Since we need the data of a stock, we will import yahoo finance libraries and retrieve the data of Tesla Inc. for this example.

The code is as follows:

# Importing libraries
import yfinance as yf
import pandas as pd

# Import Tesla data
tesla = yf.download(‘TSLA’,’2020-03-01′, ‘2020-03-30’)
tesla[‘daily_returns’] = tesla[‘Close’].pct_change()

Now, we will use the accumulate function.

# accumulate function
tesla[‘daily_returns’].dropna(inplace=True)
result = itertools.accumulate(tesla[‘daily_returns’], operator.add)
for each in result:
print(each)

In the next installment, the author will discuss the chain() iterator.

Visit https://www.quantinsti.com/ for ready-to-use Python functions as applied in trading and data analysis.

Disclosure: Interactive Brokers

Information posted on IBKR Campus that is provided by third-parties does NOT constitute a recommendation that you should contract for the services of that third party. Third-party participants who contribute to IBKR Campus are independent of Interactive Brokers and Interactive Brokers does not make any representations or warranties concerning the services offered, their past or future performance, or the accuracy of the information provided by the third party. Past performance is no guarantee of future results.

This material is from QuantInsti and is being posted with its permission. The views expressed in this material are solely those of the author and/or QuantInsti and Interactive Brokers is not endorsing or recommending any investment or trading discussed in the material. This material is not and should not be construed as an offer to buy or sell any security. It should not be construed as research or investment advice or a recommendation to buy, sell or hold any security or commodity. This material does not and is not intended to take into account the particular financial conditions, investment objectives or requirements of individual customers. Before acting on this material, you should consider whether it is suitable for your particular circumstances and, as necessary, seek professional advice.

Disclosure: Displaying Symbols on Video

Any stock, options or futures symbols displayed are for illustrative purposes only and are not intended to portray recommendations.