Как заменить все значения xarray на константу (скажем, 5) в Python?

Question

Я знаю, как заменить значения на основе условия, такого как здесь , используя .where, но я хотел бы знать, как заменить все значения в xarray на число.

Заранее спасибо

Answer 1

Из документации здесь все, что вам нужно сделать, это:

a = xr.DataArray(np.arange(16).reshape(4, 4), dims=['x', 'y'])

b = a.where(a > 5, other=5)
b = b.where(b < 5, other=5)

Поскольку r-beer предлагает прямой подход, будет:

a = a.where(a == 5, other=5)

Answer 2

Возможно, будет проще использовать следующий код:

a = xr.DataArray(np.arange(16).reshape(4, 4), dims=['x', 'y'])

c = a.where(a == 5, other=5)