/var/folders/4l/5bp8d9ks6v15tkhzpbszxxpc0000gn/T/ipykernel_21493/2296589567.py:28: UserWarning: This figure includes Axes that are not compatible with tight_layout, so results might be incorrect.
  plt.tight_layout(rect=[0, 0, 0.9, 1])  # Leave space for color bar
