added duration plot
This commit is contained in:
parent
99daa55e68
commit
cf467c8554
1 changed files with 59 additions and 0 deletions
59
duration_comparison.py
Normal file
59
duration_comparison.py
Normal file
|
|
@ -0,0 +1,59 @@
|
|||
try:
|
||||
from GC_Wrapper import GC_wrapper as GC
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
import pathlib
|
||||
import plotly
|
||||
import plotly.graph_objs as go
|
||||
import tempfile
|
||||
import pandas as pd
|
||||
|
||||
# Define temporary file
|
||||
temp_file = tempfile.NamedTemporaryFile(mode="w+t", prefix="GC_", suffix=".html", delete=False)
|
||||
|
||||
def get_time_moving(season_metrics):
|
||||
df = pd.DataFrame(season_metrics)[['date', 'Time_Moving']]
|
||||
# sum up moving times for days with multiple activities
|
||||
df = df.groupby(['date']).sum()
|
||||
df = df.cumsum()
|
||||
df = pd.DataFrame({'date': df.index, 'hours': df['Time_Moving'] / 3600.0})
|
||||
start_date = df['date'][0]
|
||||
df['days_since'] = df['date'].apply(lambda x: (x - start_date).days)
|
||||
return df
|
||||
|
||||
def cumulated_time_trace(season_metrics, color, name):
|
||||
df = get_time_moving(season_metrics)
|
||||
return go.Scatter(x=df['days_since'], y=df['hours'], line={'color': color}, name=name)
|
||||
|
||||
def main():
|
||||
season_metrics_all = GC.seasonMetrics(compare=True)
|
||||
seasons_info = GC.season(compare=True)
|
||||
|
||||
|
||||
fig = go.Figure()
|
||||
for season, name in zip(season_metrics_all, seasons_info['name']):
|
||||
color = season[1]
|
||||
season_metrics = season[0]
|
||||
trace = cumulated_time_trace(season_metrics, color, name)
|
||||
fig.add_trace(trace)
|
||||
|
||||
fig.update_layout(
|
||||
title="Duration Comparison",
|
||||
paper_bgcolor='rgba(0, 0, 0, 0)',
|
||||
plot_bgcolor='rgba(0, 0, 0, 0)',
|
||||
showlegend=True,
|
||||
xaxis=dict(showline=True, mirror=True, ticks='inside'),
|
||||
yaxis=dict(showline=True, mirror=True, ticks='inside'),
|
||||
xaxis_title='Days',
|
||||
yaxis_title='Time (h)',
|
||||
)
|
||||
fig.update_xaxes(showline=True, linewidth=1, linecolor='black', mirror=True)
|
||||
fig.update_yaxes(showline=True, linewidth=1, linecolor='black', mirror=True, showgrid=True, gridcolor='lightgray')
|
||||
plotly.offline.plot(fig, auto_open=False, filename=temp_file.name)
|
||||
GC.webpage(pathlib.Path(temp_file.name).as_uri())
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Loading…
Reference in a new issue