Deltares-research / kenmerkendewaarden

Derive indicators from waterlevel measurements
https://deltares-research.github.io/kenmerkendewaarden/
GNU General Public License v3.0
2 stars 0 forks source link

Create pcolormesh figure for data_amount data #15

Closed veenstrajelmer closed 5 months ago

veenstrajelmer commented 6 months ago

Example code:

import pandas as pd
import matplotlib.pyplot as plt
plt.close("all")
import numpy as np

df = pd.read_csv(r"p:\11210325-005-kenmerkende-waarden\work\measurements_wl_18700101_20240101\data_amount_ts.csv")
df = df.set_index("Groeperingsperiode")
df[df==0] = np.nan
df_relative = df.div(df.median(axis=1), axis=0)
df_relative = df_relative.clip(upper=2)

# df = df_relative # this is useful for ts, because the frequency was changed from hourly to 10-minute

fig, ax = plt.subplots(figsize=(14,8))
pc = ax.pcolormesh(df.columns, df.index, df.values) # , cmap="turbo")
fig.colorbar(pc, ax=ax)
ax.tick_params(axis='x', labelrotation=90)
ax.grid(zorder=-5)
fig.tight_layout()