向xarray.DataArray添加新坐標

Question

假設以下代碼具有一個xarray.DataArray ，其中包含兩個維度和一個坐標：

import numpy as np
from xarray import DataArray

data = np.random.rand(10, 4)
f_names = ['a', 'b', 'c', 'd']
sample_weights = np.random.rand(10)
rows = list(range(len(data)))
coords={'samples': rows,
        'features': f_names,
        'sample_weights': ('samples', sample_weights)}
xdata = DataArray(data, coords=coords,
                  dims=['samples', 'features'])

subset = xdata[::2]

現在，我想向subset添加另一個坐標，例如alternate_sample_weights 。 我嘗試：

subset.assign_coords(alternate_sample_weights=np.zeros(5)

導致以下錯誤：

ValueError: cannot add coordinates with new dimensions to a DataArray

API文檔非常稀疏，不確定我在做什么錯。

Answer 1

似乎在添加新坐標時，還需要傳遞添加坐標的尺寸。 因此，在這種情況下，它將是：

subset.assign_coords(
    alternate_sample_weights=('samples', np.zeros(5)))