Set coordinates of xarray DataArray dimension

Question

I like to concat/stack different DataArrays into a new one, with certain coordinates at the concat dimension:

x1 = xr.DataArray(np.arange(0,3)[...,np.newaxis], coords=[('b', np.arange(3,6)),('a', [10])]).squeeze()
x2 = xr.DataArray(np.arange(1,4)[...,np.newaxis], coords=[('b', np.arange(3,6)),('a', [10])]).squeeze()

xc = xr.concat([x1,x2], dim='new')

<xarray.DataArray (new: 2, b: 3)>
array([[0, 1, 2],
       [1, 2, 3]])
Coordinates:
  * b        (b) int64 3 4 5
    a        int64 10
Dimensions without coordinates: new

How can I set the coordinates of dimension with the concat function 'new'?

jhamman · Accepted Answer · 2019-02-04 16:28:11Z

2

The dim argument in the concat function can take a DataArray or pandas.Index in addition to the name of the concat dimension:

xc = xr.concat([x1,x2], dim=xr.DataArray(['a', 'b'], dims='new', name='new'))

More detail in the xarray.concat documentation: http://xarray.pydata.org/en/stable/generated/xarray.concat.html#xarray.concat

answered Feb 4, 2019 at 16:28

jhamman

6,5441 gold badge26 silver badges44 bronze badges

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

Comments

Vasco · Accepted Answer · 2019-02-04 15:50:51Z

1

Setting the coordinates can be done via:

xc['new'] = ['a', 'b']

answered Feb 4, 2019 at 15:50

Vasco

1,19711 silver badges28 bronze badges

2 Comments

Vasco Over a year ago

This needs two line of codes, can it be done in one?

Steve Over a year ago

Sometimes the solution with more lines of code is easier to read. I am much more likely to remember this solution than the more complex one-liner.

Collectives™ on Stack Overflow

Set coordinates of xarray DataArray dimension

2 Answers 2

Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related