mirror
/
telegram-stats-bot
kopia lustrzana https://github.com/mkdryden/telegram-stats-bot
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

stats: fix hours plots when time period under 24 hours

pull/1/head
Michael DM Dryden 2020-06-05 02:17:26 -04:00
rodzic cfa6591408
commit 2e574f8a07
1 zmienionych plików z 5 dodań i 1 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

6
telegram_stats_bot/stats.py
Wyświetl plik

@ -208,6 +208,10 @@ class StatsRunner(object):
df['day'] = df.day.dt.tz_convert(self.tz) df['day'] = df.day.dt.tz_convert(self.tz)
df = df.set_index('day') df = df.set_index('day')
df = df.asfreq('h', fill_value=0) # Insert 0s for periods with no messages df = df.asfreq('h', fill_value=0) # Insert 0s for periods with no messages
if (df.index.max() - df.index.min()) < pd.Timedelta('24 hours'): # Deal with data covering < 24 hours
df = df.reindex(pd.date_range(df.index.min(), periods=24, freq='h'))
df['hour'] = df.index.hour df['hour'] = df.index.hour
if user: if user:
@ -225,7 +229,7 @@ class StatsRunner(object):
subplot.set_ylim(bottom=0, top=df['messages'].quantile(0.999, interpolation='higher')) subplot.set_ylim(bottom=0, top=df['messages'].quantile(0.999, interpolation='higher'))
subplot.axvspan(11.5, 23.5, zorder=0, color=(0, 0, 0, 0.05)) subplot.axvspan(11.5, 23.5, zorder=0, color=(0, 0, 0, 0.05))
subplot.set_xlim(-1, 24) subplot.set_xlim(-1, 24) # Set explicitly to plot properly even with missing data
if user: if user:
subplot.set_ylabel('Messages per Week') subplot.set_ylabel('Messages per Week')