1234567891011121314151617181920212223242526272829303132333435 |
- import pickle
- import pandas as pd
- import quandl
- api_key = 'rFsSehe51RLzREtYhLfo'
- def state_list():
- fifty_states = pd.read_html('https://simple.wikipedia.org/wiki/List_of_U.S._states')
- return fifty_states[0][0][1:]
- def initial_state_data():
- states = state_list()
- main_df = pd.DataFrame()
- for abbv in states:
- query = 'FMAC/HPI_' + str(abbv)
- df = quandl.get(query, authtoken=api_key)
- df.columns = [str(abbv)]
- if main_df.empty:
- main_df = df
- else:
- main_df = main_df.join(df)
- print(main_df.head())
- pickle_out = open('fifty_states.pickle', 'wb')
- pickle.dump(main_df, pickle_out)
- pickle_out.close()
- # initial_state_data()
- pickle_in = open('fifty_states.pickle' , 'rb')
- HPI_data = pickle.load(pickle_in)
- print(HPI_data)
|