@@ -27,14 +27,15 @@ def get_before_2016_02(tickers, store_dir):
2727 start = dt .date (2009 ,1 ,1 )
2828 end = dt .date (2016 ,1 ,31 )
2929 CLOSE_FIELD = 'Close'
30+ ADJUSTED_CLOSE_FIELD = 'Adj Close'
3031 for ticker in tickers :
3132 stock_price_filename = store_dir + 'close_prices_pre_2016_02_' + ticker .upper () + '.csv'
3233 print 'Saving to' + stock_price_filename
3334 time .sleep (1 )
3435 try :
35- ticker_df = web .DataReader (ticker , 'google ' , start , end )
36+ ticker_df = web .DataReader (ticker , 'yahoo ' , start , end )
3637 # drop rows with at least 1 NaN
37- df_to_save = ticker_df [[CLOSE_FIELD ]].dropna (thresh = 1 )
38+ df_to_save = ticker_df [[CLOSE_FIELD , ADJUSTED_CLOSE_FIELD ]].dropna (thresh = 1 )
3839 df_to_save .to_csv (stock_price_filename )
3940 except Exception , e :
4041 print 'Failed ticker: ' + ticker + ' ~~~' , str (e )
@@ -45,18 +46,19 @@ def get_after_2016_02(tickers, store_dir):
4546 start = dt .date (2016 ,2 ,1 )
4647 end = dt .date .today ()
4748 CLOSE_FIELD = 'Close'
49+ ADJUSTED_CLOSE_FIELD = 'Adj Close'
4850 for ticker in tickers :
4951 stock_price_filename = store_dir + 'close_prices_post_2016_02_' + ticker .upper () + '.csv'
5052 print 'Saving to' + stock_price_filename
5153 time .sleep (1 )
5254 try :
53- ticker_df = web .DataReader (ticker , 'google ' , start , end )
55+ ticker_df = web .DataReader (ticker , 'yahoo ' , start , end )
5456 # drop rows with at least 1 NaN
55- df_to_save = ticker_df [[CLOSE_FIELD ]].dropna (thresh = 1 )
57+ df_to_save = ticker_df [[CLOSE_FIELD , ADJUSTED_CLOSE_FIELD ]].dropna (thresh = 1 )
5658 df_to_save .to_csv (stock_price_filename )
5759 except Exception , e :
5860 print 'Failed ticker: ' + ticker + ' ~~~' , str (e )
5961
6062
61- get_before_2016_02 (merge_lists (),stock_prices_dir )
63+ # get_before_2016_02(merge_lists(),stock_prices_dir)
6264get_after_2016_02 (merge_lists (),stock_prices_dir )
0 commit comments