xarray_dataarray_creation_subsetting_and_aggregation_quickstart.py

1import numpy as np
2import pandas as pd
3import xarray as xr
4
5# Create a DataArray from a numpy array
6data = np.random.rand(4, 3)
7locs = ["IA", "IL", "IN"]
8times = pd.date_range("2000-01-01", periods=4)
9
10foo = xr.DataArray(data, coords=[times, locs], dims=["time", "space"])
11
12# Operations and subsetting
13# Selection by label
14selected = foo.sel(time="2000-01-01")
15
16# Vectorized operations
17result = foo * 10
18
19# Aggregations along dimensions
20mean_values = foo.mean(dim="time")
21
22# Convert to/from pandas
23df = foo.to_series()
24back_to_xr = xr.DataArray.from_series(df)
25
26# Plotting (requires matplotlib)
27# foo.plot()
28
29print(foo)