設置或更改新 xarray 維度的值

Question

我有一個帶有藍色波段圖像（形狀 600 x 600）的 xarray DataArray，我想將其轉換為偽 RGB 圖像（即在 RGB 模式下查看時將顯示為藍色的圖像）。 我的想法是，如果我添加兩個波段（紅色和綠色）並將它們的值設置為 0，就可以解決問題（我使用 rioxarray 將數組保存為 PNG）。

dummy_image = np.random.randint(0, 255, size=(600, 600))
im_xa_band = xa.DataArray(dummy_image, dims=['y', 'x'], coords={'x': np.arange(0, 600), 'y': np.arange(0, 600)})
im_xa_3band = im_xa.expand_dims({'band': 3})
im_xa_3band = im_xa_3band.assign_coords(band=['R', 'G', 'B'])

當我這樣做時，兩個新波段獲得與原始藍色波段相同的值。 但是，我沒有找到將新波段設置為 0 的方法（或者更確切地說，如何設置它們的值，似乎是一項如此簡單的任務）。

.sel() 不能用於賦值。 當我嘗試 Pandas 中的 try.loc 時，出現錯誤：

im_xa_3band.loc['R', :, :] = 0

ValueError: Assignment destination is a view. Do you want to .copy() array first?

任何有關如何更改值的建議都將受到贊賞。

Answer 1

看來您需要在.copy() .expand_dims() ，請參閱此處的討論。

im_xa_3band = im_xa.expand_dims({'band': 3}).copy()
im_xa_3band = im_xa_3band.assign_coords(band=['R', 'G', 'B'])
im_xa_3band.loc['R', :, :] = 0