From bd0ca0ea4c9b51f203937e2c2f9fca6d52d882b1 Mon Sep 17 00:00:00 2001 From: Tobias Date: Wed, 16 Sep 2020 09:54:02 +0200 Subject: [PATCH] minor changes to main --- main.py | 22 ++++++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index ccde0f0..ab8ae52 100644 --- a/main.py +++ b/main.py @@ -1,14 +1,32 @@ +import os + +import pandas import yfinance as yf from get_all_tickers.get_tickers import get_tickers +PROJ_PATH = os.path.dirname(__file__) +ticker_data_file = os.path.join(PROJ_PATH, "ticker_data_NYSE.csv") + def main(): - all_tickers = get_tickers() + all_tickers = get_tickers(NASDAQ=False, AMEX=False) print(all_tickers) data = yf.download(tickers="SPL AAPL", start="2017-01-01", end="2017-04-30", interval="60m", ) + # data = yf.download(tickers="SPL AAPL", start="2017-01-01", end="2017-04-30", interval="1d", ) + data = yf.download(tickers=" ".join(all_tickers), interval="1d") + # yf.download() + print(data) + print(type(data)) + + # data.to_csv(ticker_data_file) + + +def transform_data(): + data = pandas.read_csv(ticker_data_file) if __name__ == '__main__': - main() + # main() + transform_data()