from pathlib import Path

import matplotlib
import matplotlib.pyplot as plt
import numpy as np


output = Path("colorbar-add.png")

measurements = np.array(
    [
        [18, 20, 23, 26, 28],
        [17, 21, 24, 27, 30],
        [16, 19, 22, 25, 29],
        [15, 18, 21, 24, 27],
    ]
)

fig, ax = plt.subplots(figsize=(6, 3.8), layout="constrained")
image = ax.imshow(measurements, cmap="viridis", vmin=15, vmax=30)

ax.set_title("Sensor temperature by rack")
ax.set_xlabel("Rack column")
ax.set_ylabel("Rack row")
ax.set_xticks(range(measurements.shape[1]))
ax.set_yticks(range(measurements.shape[0]))

colorbar = fig.colorbar(image, ax=ax, label="Temperature (deg C)")

fig.savefig(output, dpi=160)
plt.close(fig)

print(f"matplotlib {matplotlib.__version__}")
print(f"colorbar label: {colorbar.ax.get_ylabel()}")
print(f"colorbar orientation: {colorbar.orientation}")
print(f"saved: {output}")
print(f"bytes: {output.stat().st_size}")