python - 如何获取xarray中最大值的坐标？

Question

一个简单的问题：我不仅想要最大值，而且还希望它在xarray DataArray中的坐标。怎么做？

我当然可以编写自己的简单reduce函数，但是我想知道xarray中是否内置了什么？

Answer 1

您可以使用da.where()根据最大值进行过滤：

In [17]: da = xr.DataArray(
             np.random.rand(2,3), 
             dims=list('ab'), 
             coords=dict(a=list('xy'), b=list('ijk'))
         )

In [18]: da.where(da==da.max(), drop=True).squeeze()
Out[18]:
<xarray.DataArray ()>
array(0.96213673)
Coordinates:
    a        <U1 'x'
    b        <U1 'j'

编辑：更新示例以更清楚地显示索引，因为xarray没有默认索引