claudecode/AvianVisitors
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix daily_plot could show zeros when max count/h is only one

Browse Source
This commit is contained in:
frederik
2024-04-05 17:30:41 +02:00
parent 078d394f9e
commit 23a35a289c
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/daily_plot.py
+3
View File
@@ -7,6 +7,7 @@ from time import sleep
import matplotlib.font_manager as font_manager import matplotlib.font_manager as font_manager
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np
import pandas as pd import pandas as pd
import seaborn as sns import seaborn as sns
from matplotlib import rcParams from matplotlib import rcParams
@@ -88,6 +89,8 @@ def create_plot(df_plt_today, now, is_top):
hours_in_day = pd.Series(data=range(0, 24)) hours_in_day = pd.Series(data=range(0, 24))
heat_frame = pd.DataFrame(data=0, index=heat.index, columns=hours_in_day) heat_frame = pd.DataFrame(data=0, index=heat.index, columns=hours_in_day)
heat = (heat+heat_frame).fillna(0) heat = (heat+heat_frame).fillna(0)
# mask out zeros, so they do not show up in the final plot. this happens when max count/h is one
heat[heat == 0] = np.nan
# Generatie heatmap plot # Generatie heatmap plot
plot = sns.heatmap(heat, norm=LogNorm(), annot=True, annot_kws={"fontsize": 7}, fmt="g", cmap=pal, square=False, plot = sns.heatmap(heat, norm=LogNorm(), annot=True, annot_kws={"fontsize": 7}, fmt="g", cmap=pal, square=False,