|
@@ -14,8 +14,6 @@ def get_data(query_manager: QueryManager, unit=None, **kwargs) -> pd.DataFrame:
|
|
|
d = pd.DataFrame(query_manager.get_historic_prices_data(unit, **kwargs))
|
|
|
if d.empty:
|
|
|
return d
|
|
|
- print(d.info())
|
|
|
- print(d)
|
|
|
d['ts_month'] = d['ts_raw'].apply(lambda x: date(x.date().year, x.date().month,1))
|
|
|
d[['price','quantity']] = d[['price','quantity']].apply(
|
|
|
lambda y: y.apply(lambda x: x and float(x)),
|
|
@@ -25,8 +23,6 @@ def get_data(query_manager: QueryManager, unit=None, **kwargs) -> pd.DataFrame:
|
|
|
def pivot_data(data: QueryManager):
|
|
|
pivot = data.groupby(['ts_month','group',])['price', 'quantity'].sum()
|
|
|
pivot = pivot.reset_index().set_index('group')
|
|
|
- print(pivot.info())
|
|
|
- print(pivot)
|
|
|
return pivot
|
|
|
|
|
|
def pie(p: pd.DataFrame, col=None, title=None):
|