/var/folders/4l/5bp8d9ks6v15tkhzpbszxxpc0000gn/T/ipykernel_21493/806804362.py:29: 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
