Python - Concat Список Dataframes с использованием DataReader - PullRequest
Новая
       10

Python - Concat Список Dataframes с использованием DataReader

0 голосов
/ 24 декабря 2018

Я пытаюсь получить данные об акциях, используя pandas_datareader.Учитывая несколько стандартных символов, достаточно просто вручную составить сценарий каждого вызова DataReader и затем согласовать результат.Однако, если длина списка увеличивается до гораздо большего числа, должен быть более простой способ итерации процесса. 

import pandas_datareader.data as web
import pandas as pd

symbols = ['AMZN','AAPL','MSFT','NFLX','GOOGL']
for i in symbols:
    print(i)

dfAMZN = web.DataReader('AMZN','yahoo')
dfAMZN['Symbol'] = 'AMZN'
dfAMZN['Date'] = dfAMZN.index

dfAAPL = web.DataReader('AAPL','yahoo')
dfAAPL['Symbol'] = 'AAPL'
dfAAPL['Date'] = dfAAPL.index

dfMSFT = web.DataReader('MSFT','yahoo')
dfMSFT['Symbol'] = 'MSFT'
dfMSFT['Date'] = dfMSFT.index

dfNFLX = web.DataReader('NFLX','yahoo')
dfNFLX['Symbol'] = 'NFLX'
dfNFLX['Date'] = dfNFLX.index

dfGOOGL = web.DataReader('GOOGL','yahoo')
dfGOOGL['Symbol'] = 'GOOGL'
dfGOOGL['Date'] = dfGOOGL.index

frames = [dfAMZN, dfAAPL, dfMSFT, dfNFLX, dfGOOGL]
dfStocks = pd.concat(frames)

Есть ли способ перебрать список символов и выполнитьшаги ниже, вместо того, чтобы просто напечатать i?

1 Ответ

0 голосов
/ 24 декабря 2018

Используйте list comprehension с assign для новых столбцов: 

symbols = ['AMZN','AAPL','MSFT','NFLX','GOOGL']
frames = [web.DataReader(i,'yahoo').assign(Symbol = i, Date = lambda x: x.index) 
          for i in symbols]

dfStocks = pd.concat(frames)

Другая альтернатива: 

symbols = ['AMZN','AAPL','MSFT','NFLX','GOOGL']

frames = []
for i in symbols:
    df = web.DataReader(i,'yahoo')
    df['Symbol'] = i
    df['Date'] = df.index
    frames.append(df)

dfStocks = pd.concat(frames)
...