Как обновить `xarray.DataArray` с помощью`.sel()` indexer?
Я нашел самый простой способ визуализировать создание N-dimensional
DataArray
должен был сделать np.ndarray
и затем заполните значения по координатам, которые я создал. Когда я попытался на самом деле сделать это, я не мог понять, как обновить xr.DataArray
,
Как я могу обновить xr.DataArray
Я инициализировал, используя метки, которые я создал? Мои фактические данные - намного более сложный набор данных, но это подводит итог тому, что я пытаюсь сделать. я могу использовать .loc
но иногда мой ndarrays
становиться огромным и сложным, где я действительно не знаю порядок тусклых тем.
# Construct DataArray
DA = xr.DataArray(np.ndarray((3,3,5)), dims=["axis_A","axis_B","axis_C"], coords={"axis_A":["A_%d"%_ for _ in range(3)],
"axis_B":["B_%d"%_ for _ in range(3)],
"axis_C":["C_%d"%_ for _ in range(5)]})
# <xarray.DataArray (axis_A: 3, axis_B: 3, axis_C: 5)>
# array([[[ 0., 0., 0., 0., 0.],
# [ 0., 0., 0., 0., 0.],
# [ 0., 0., 0., 0., 0.]],
# [[ 0., 0., 0., 0., 0.],
# [ 0., 0., 0., 0., 0.],
# [ 0., 0., 0., 0., 0.]],
# [[ 0., 0., 0., 0., 0.],
# [ 0., 0., 0., 0., 0.],
# [ 0., 0., 0., 0., 0.]]])
# Coordinates:
# * axis_B (axis_B) <U3 'B_0' 'B_1' 'B_2'
# * axis_A (axis_A) <U3 'A_0' 'A_1' 'A_2'
# * axis_C (axis_C) <U3 'C_0' 'C_1' 'C_2' 'C_3' 'C_4'
# # Update?
DA.sel(axis_A="A_1", axis_B="B_1", axis_C="C_1").values = 1
DA.max()
# # <xarray.DataArray ()>
# # array(0.0)
DA.sel(axis_A="A_1", axis_B="B_1", axis_C="C_1") = 1
# # File "<ipython-input-17-8feb7332633f>", line 4
# # DA.sel(axis_A="A_1", axis_B="B_1", axis_C="C_1") = 1
# # ^
# # SyntaxError: can't assign to function call
1 ответ
Это действительно неудобно из-за неудачного ограничения синтаксиса Python, что аргументы ключевого слова не поддерживаются в квадратных скобках.
Таким образом, вместо этого вам нужно поставить аргументы .sel
как словарь в .loc
вместо:
DA.loc[dict(axis_A="A_1", axis_B="B_1", axis_C="C_1")] = 1