Browse Source

add label to plots

Daniel Sheffield 2 years ago
parent
commit
eacbbca5f7
1 changed files with 5 additions and 2 deletions
  1. 5 2
      app/activities/Plot.py

+ 5 - 2
app/activities/Plot.py

@@ -40,11 +40,14 @@ for k in ('group', 'category', 'product', 'unit'):
 fields = dict((k,v or None) for k,v in fields.items())
 
 print(fields)
-d = query_manager.get_historic_prices_data(fields['unit'] or 'kg', **dict((k,v) for k,v in fields.items() if k != 'unit'))
+unit = fields['unit'] or 'kg'
+d = query_manager.get_historic_prices_data(unit, **dict((k,v) for k,v in fields.items() if k != 'unit'))
 sns.set_theme()
 d = d.pivot_table(index=['ts_raw',], columns=['product',], values=['$/unit'], aggfunc='mean')
 d.columns = d.columns.droplevel()
 print(d.info())
 print(d)
-sns.lineplot(data=d, markers=True)
+ax = sns.lineplot(data=d, markers=True)
+ax.set_xlabel('Time')
+ax.set_ylabel(f'$ / {unit}')
 plt.show()