python - xarrayで最大値の座標を取得するには?

Question

簡単な質問: xarray DataArray 内の最大値だけでなく、その座標も必要です。どうやってするか？

もちろん、独自の単純な reduce 関数を作成することもできますが、xarray に組み込まれているものはあるのでしょうか?

Answer 1

スタックを使用することもできます:

data が時間、経度、緯度を含む 3D 変数であり、経時的な最大値の座標が必要だとします。

stackdata = data.stack(z=('lon', 'lat'))
maxi = stackdata.argmax(axis=1)
maxipos = stackdata['z'][maxi]
lonmax = [maxipos.values[itr][0] for itr in range(ntime)]
latmax = [maxipos.values[itr][1] for itr in range(ntime)]