add new coordinates to an xarray.DataArray

Question

Assume the following code, which has an xarray.DataArray with two dimensions and a coordinate:

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]

Now I want to add another coordinates, like alternate_sample_weights to subset. I try:

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

which results in the following error:

ValueError: cannot add coordinates with new dimensions to a DataArray

The API documentation is pretty sparse, not sure what I'm doing wrong.

adrin · Accepted Answer · 2019-07-09 11:33:28Z

5

It appears that when adding a new coordinate, you need to pass also the dimension along which it's added. So in this case, it would be:

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

answered Jul 9, 2019 at 11:33

adrin

4,9665 gold badges38 silver badges54 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

add new coordinates to an xarray.DataArray

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related