xarray_dataset_creation_coordinate_selection_and_mean_computation.py

1import numpy as np
2import pandas as pd
3import xarray as xr
4
5# Create some sample data
6data = np.random.randn(2, 3)
7locs = ["IA", "IL", "IN"]
8times = pd.date_range("2000-01-01", periods=2)
9
10# Initialize an xarray.DataArray
11foo = xr.DataArray(data, coords=[times, locs], dims=["time", "space"])
12
13# Combine DataArrays into a Dataset
14ds = xr.Dataset({"foo": foo, "bar": ("x", [1, 2, 3])})
15
16# Basic operations: selection by label
17subset = ds.sel(time="2000-01-01")
18
19# Computation: mean across a specific dimension
20mean_val = ds.mean(dim="time")
21
22print(ds)
23print("\nMean across time:")
24print(mean_val)