查找满足条件的xarray索引

时间:2020-04-06 00:13:15

标签: python python-xarray

我想获取满足某些条件的xarray数据数组的索引。在相关线程(here)中提供的关于如何找到最大位置的答案对我也不起作用。就我而言,我也想找出其他类型条件的位置,而不仅仅是最大条件。这是我尝试过的:

import pandas as pd
a  = [ {111: [1, 2, 3]}, {112:[4, 5, 6]}, {113:[7,8,9]}]

for item in a:
    for k, v in item.items():
        df[k] = v

df

    111 112 113
0   1   4   7
1   2   5   8
2   3   6   9

即使执行相同的命令,也不会返回我所链接的示例中所示的位置。不知道为什么。

1 个答案:

答案 0 :(得分:1)

我更新了链接的示例,以更清楚地显示索引。由于xarray不再添加默认索引,因此上一个示例找到了最大位置,但未显示索引。复制如下:

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'