sortion/BirdNET-stream
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

plotter: Fix color map indexing for counterplot

Browse Source
This commit is contained in:
Samuel Ortion 2022-08-21 09:07:16 +02:00
parent 0cfa8715c2
commit ec09401396
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
daemon/plotter/chart.py
View File

@ -11,7 +11,7 @@ CONFIG = {
"readings": 10, "readings": 10,
"palette": "Greens", "palette": "Greens",
"db": "./var/db.sqlite", "db": "./var/db.sqlite",
"date": "2022-08-14" "date": datetime.now().strftime("%Y-%m-%d")
} }
db = sqlite3.connect(CONFIG['db']) db = sqlite3.connect(CONFIG['db'])
@ -27,7 +27,10 @@ df['date'] = df['date'].astype(str)
df_on_date = df[df['date'] == CONFIG['date']] df_on_date = df[df['date'] == CONFIG['date']]
top_on_date = (df_on_date['common_name'].value_counts()[:CONFIG['readings']]) top_on_date = (df_on_date['common_name'].value_counts()[:CONFIG['readings']])
if top_on_date.empty:
print("No observations on {}".format(CONFIG['date']))
exit()
df_top_on_date = df_on_date[df_on_date['common_name'].isin(top_on_date.index)] df_top_on_date = df_on_date[df_on_date['common_name'].isin(top_on_date.index)]
# Create a figure with 2 subplots # Create a figure with 2 subplots
@ -37,7 +40,7 @@ plt.subplots_adjust(left=None, bottom=None, right=None,
top=None, wspace=0, hspace=0) top=None, wspace=0, hspace=0)
# Get species frequencies # Get species frequencies
frequencies_order = pd.value_counts(df['common_name']).iloc[:CONFIG['readings']].index frequencies_order = pd.value_counts(df_top_on_date['common_name']).iloc[:CONFIG['readings']].index
# Get min max confidences # Get min max confidences
confidence_minmax = df_top_on_date.groupby('common_name')['confidence'].max() confidence_minmax = df_top_on_date.groupby('common_name')['confidence'].max()
confidence_minmax = confidence_minmax.reindex(frequencies_order) confidence_minmax = confidence_minmax.reindex(frequencies_order)
@ -84,8 +87,8 @@ for _, spine in plot.spines.items():
plot.set(ylabel=None) plot.set(ylabel=None)
plot.set(xlabel="Hour of day") plot.set(xlabel="Hour of day")
fig.subplots_adjust(top=0.9) fig.subplots_adjust(top=0.9)
plt.suptitle(f"Top {CONFIG['readings']} species (Updated on {datetime.now().strftime('%Y/%m-%d %H:%M')})") plt.suptitle(f"Top {CONFIG['readings']} species on {CONFIG['date']}", fontsize=14)
plt.title(f"(Updated on {datetime.now().strftime('%Y/%m-%d %H:%M')})")
plt.savefig(f"./var/charts/chart_{CONFIG['date']}.png", dpi=300) plt.savefig(f"./var/charts/chart_{CONFIG['date']}.png", dpi=300)
plt.close() plt.close()