Expand Coordinate by Concat with other DataArray

[babameme]

Hi team,
I have this problem, can you help me to solve it:

import xarray as xr
import numpy as np
import pandas as pd
ds1 = xr.DataArray(np.arange(60).reshape(12, 5), coords = {'X': range(12), 'Y': range(5)}, dims = ['X', 'Y'])
ds2 = xr.DataArray(np.arange(60).reshape(12, 5), coords = {'X': range(12), 'Y': range(5)}, dims = ['X', 'Y'])
ds3 = xr.DataArray(np.arange(60).reshape(12, 5), coords = {'X': range(12), 'Y': range(5)}, dims = ['X', 'Y'])
ds = xr.concat([ds1, ds2, ds3], pd.Index(['feature_0', 'feature_1', 'feature_2'], name = 'features'))
one_xr = xr.full_like(ds1, 1.)

I want to concat/combine/merge one_xr with ds along 'X' and 'Y' dims.
The result i expect is a new DataArray with:

• dims = ['X', 'Y', 'features']
• coords = {'X': range(12), 'Y': range(5), 'features': ['constant', 'feature_0', 'feature_1', 'feature_2']}

Thank
Regards

[max-sixty]

Could you edit your question such that it's possible to copy & paste the example?

[babameme]

Hi @max-sixty , i have edited it.

[max-sixty]

There may be an easier way, but this works:

In [10]: xr.concat([ds, one_xr.expand_dims('features')], dim='features')
Out[10]:
<xarray.DataArray (features: 4, X: 12, Y: 5)>
array([[[ 0,  1,  2,  3,  4],
        [ 5,  6,  7,  8,  9],
        [10, 11, 12, 13, 14],
        [15, 16, 17, 18, 19],
        [20, 21, 22, 23, 24],

(Note that ds is a DataArray)

[babameme]

Thanks @max-sixty , it works