在 Python 中使用 xarray 的其他兩個變量擴展變量的坐標

Question

假設我有 16 個網格單元格 (4 * 4) 的數據，每個單元格都有一個相應的索引 (0~15) 作為維度和坐標以及變量（ a ，經度和緯度）。 這是創建此數據的代碼。

import xarray as xr
import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(16,3), \
                  columns=['a','longitude', 'latitude'], \
                  index=range(16))
ds = df.to_xarray()
ds

我想做的是：

使用每個單元格的經度和緯度變量將數據a從 ( index ) 擴展到 ( longitude , latitude ) 的協調。
因此，生成的 DataSet 將包括經度和緯度作為其維度和坐標以及（經度，緯度）的變量a
如何在 xarray 功能中做到這一點？
謝謝！

Answer 1

這是解決該問題的一種方法：

# convert the index as a MultiIndex containing longitude and latitude
dat_2d = ds.set_index({'index': ['longitude', 'latitude']})

# unstack the MultiIndex
unstacked = dat_2d.unstack('index')

# plot
unstacked['a'].plot()