在xarray上迭代。DataArray第一维及其坐标

Question

假设我有以下DataArray

arr = xarrray.DataArray(np.arange(6).reshape(2,3),
                        dims=['A', 'B'],
                        coords=dict(A=['a0', 'a1'], 
                                    B=['b0', 'b1', 'b2']))

我想迭代第一个维度并执行以下操作（当然，我想做比打印更复杂的事情）

^{pr2}$

然后得到

a0 [0 1 2]
a1 [3 4 5]

我不熟悉xarray，所以我想知道是否有更自然的方法来实现这一点，比如

for coor, sub_arr in arr.some_method():
    print(coor, sub_arr)