xpystac

xpystac提供将xarray.open_dataset扩展以接受pystac对象的粘合剂。

目标是只要这个库在你的环境中，你就不必担心它。

示例

搜索COGs集合

import pystac_client
import xarray as xr


catalog = pystac_client.Client.open(
    "https://earth-search.aws.element84.com/v1",
)

search = catalog.search(
    intersects=dict(type="Point", coordinates=[-105.78, 35.79]),
    collections=['sentinel-2-l2a'],
    datetime="2022-04-01/2022-05-01",
)

xr.open_dataset(search, engine="stac")

以下是一些来自行星计算机文档的示例

import planetary_computer
import pystac_client
import xarray as xr


catalog = pystac_client.Client.open(
    "https://planetarycomputer.microsoft.com/api/stac/v1",
    modifier=planetary_computer.sign_inplace,
)

从参考文件读取

collection = catalog.get_collection("nasa-nex-gddp-cmip6")
asset = collection.assets["ACCESS-CM2.historical"]

xr.open_dataset(asset)

参考: https://planetarycomputer.microsoft.com/dataset/nasa-nex-gddp-cmip6#Example-Notebook

从zarr文件读取

collection = catalog.get_collection("daymet-daily-hi")
asset = collection.assets["zarr-abfs"]

xr.open_dataset(asset)

参考: https://planetarycomputer.microsoft.com/docs/quickstarts/reading-zarr-data/

安装

pip install git+https://github.com/stac-utils/xpystac

工作原理

当你调用xarray.open_dataset(object, engine="stac")时，这个库将该open调用映射到正确的库。根据object的类型，这可能是一个堆叠库（要么是odc-stac或stackstac）或者返回到xarray.open_dataset本身，但是使用从pystac对象拉取的引擎和其他选项。

现有技术

这项工作受到了https://github.com/TomAugspurger/staccontainers以及https://github.com/stac-utils/pystac/issues/846中的讨论的启发