xarray_dataset_creation_labeling_computation_and_plotting_quickstart.py

1import xarray as xr
2import numpy as np
3import pandas as pd
4
5# Create some sample data
6data = np.random.randn(2, 3)
7dims = ("x", "y")
8coords = {"x": [10, 20], "y": [100, 101, 102]}
9
10# Initialize an xarray.DataArray
11foo = xr.DataArray(data, coords=coords, dims=dims, name="foo")
12
13# Create a Dataset by combining DataArrays
14ds = xr.Dataset({"foo": foo, "bar": ("x", [1, 2])})
15
16# Indexing and selecting data
17# Selection by label
18selected = ds.sel(x=10)
19
20# Computation (automatic alignment by coordinates)
21result = ds + 10
22
23# Aggregation (mean along a specific dimension)
24mean_ds = ds.mean(dim="x")
25
26# Conversion to pandas
27df = ds.to_dataframe()
28
29# Basic plotting (requires matplotlib)
30# ds.foo.plot()
31
32print(ds)