Update README.md

README.md

@@ -96,15 +96,36 @@ Below you can find an example to read that file and load it as numpy array or xa
 ```
 
 import earthkit.data as ekd
+import cartopy.crs as ccrs
+import numpy as np
 
 source_filename='anemoi.grib'
 aifs_forecast = ekd.from_source('file',source_filename)
 
-# to load it as a numpy array
-aifs_forecast.to_numpy() #
+# to load a variable as a numpy array
+aifs_forecast.sel(param="2t").to_numpy() 
 
-# to load it as a xarray array
-aifs_forecast.to_xarray()
+# to load a variables as an xarray array
+aifs_forecast.sel(param="2t").to_xarray()
+
+aifs_forecast_2t_xr = aifs_forecast.sel(param="2t").to_xarray()
+
+fig, axes = plt.subplots(subplot_kw={'projection': ccrs.Robinson()}, figsize=(16,6))
+
+map_scatter(data_v021_2t_xr.latitude, data_v021_2t_xr.longitude, data_v021_2t_xr.t2m[0,1,0,:],
+            axes=axes, 
+            proj=ccrs.Robinson(),
+            vmin=np.percentile(data_v021_2t_xr.t2m[0,1,0,:], 2),
+            vmax=np.percentile(data_v021_2t_xr.t2m[0,1,0,:], 98),
+            cmap="RdYlBu_r",
+            mask_ocean=True,
+            plot_title='2-Meter Temperature Forecast',
+            cmap_label="Temperature (K)"
+            )
+
+<div style="display: flex; justify-content: center;">
+  <img src="2t_aifs_v021.jpeg" alt="Encoder graph" style="width: 50%;"/>
+</div>
 
 ```