td_intern rossmann

var_imp.py

%matplotlib inline

import os
import pandas as pd
import pandas_td as td
import matplotlib.pyplot as plt

con = td.connect(apikey=os.environ['TD_API_KEY'],endpoint='https://api.treasuredata.com/')
engine = con.query_engine(database='rossmann', type='presto')
var_imp=td.read_td_table('var_importance', engine)
del var_imp['time']
X=range(var_imp.shape[1])
plt.barh(X,var_imp.values[0],align="center")
plt.yticks(X,var_imp.columns)
plt.savefig('var_imp.png')