如何从DataArrayGroupBy对象中取一个或多个分组?
时间: 2023-06-20 20:08:32 浏览: 97
mysql group by 对多个字段进行分组操作
要从DataArrayGroupBy对象中获取一个或多个分组,可以使用`.get_group()`方法。该方法需要一个参数,即要获取的分组的名称或标签,可以是单个值或一个列表,返回一个新的DataArray对象,其中包含所选分组的所有数据。例如:
```python
import xarray as xr
# 创建一个 DataArray 对象
data = xr.DataArray(
[[1, 2], [3, 4], [5, 6]],
dims=("x", "y"),
coords={"x": ["a", "b", "c"], "y": ["foo", "bar"]}
)
# 按 'x' 维度进行分组
grouped = data.groupby("x")
# 获取 'a' 分组
group_a = grouped.get_group("a")
print(group_a)
# 获取 'b' 和 'c' 分组
group_bc = grouped.get_group(["b", "c"])
print(group_bc)
```
输出:
```
<xarray.DataArray (y: 2)>
array([1, 2])
Coordinates:
x <U1 'a'
* y (y) <U3 'foo' 'bar'
<xarray.DataArray (x: 2, y: 2)>
array([[3, 4],
[5, 6]])
Coordinates:
* x (x) <U1 'b' 'c'
* y (y) <U3 'foo' 'bar'
```
阅读全文