plot_rainfall.py

import numpy as np
import xarray as xr
import matplotlib.pyplot as plt
import cartopy.crs as ccrs

ds = xr.open_dataset("RF25_ind2025_rfp25.nc")
rain = ds["RAINFALL"]

rain = rain.where(rain < 9e20)

day = rain.isel(TIME=0)

fig, ax = plt.subplots(
    figsize=(8, 7),
    subplot_kw={"projection": ccrs.PlateCarree()}
)

mesh = ax.pcolormesh(
    rain.LONGITUDE, rain.LATITUDE, day,
    cmap="YlGnBu",
    vmin=0, vmax=50,
    transform=ccrs.PlateCarree(),
    shading="auto"
)
ax.coastlines(resolution="50m", linewidth=0.8)

ax.set_extent([66.5, 100.0, 6.5, 38.5], crs=ccrs.PlateCarree())

plt.colorbar(mesh, ax=ax, label="Rainfall (mm)", shrink=0.8)
ax.set_title("Rainfall — Day 1")

plt.savefig("rainfall_map.png", dpi=150, bbox_inches="tight")
plt.show()