如何在xarray/rioxarray中用变量或列表元素指定波段？

我遍历NetCDF文件时，已通过匹配预期波段列表与文件可用波段，得到共同波段名称列表common_key。但用列表元素替代硬编码波段名调用函数时出错：

• 正常运行代码：bandcount = up_sampled.band_name.shape[0]
• 修改后报错代码：bandcount = up_sampled.common_key[0].shape[0]
• 错误提示：'Dataset' object has no attribute 'common_key'

需要实现类似字符串替换的波段名称代入功能，匹配波段的代码如下：

# 用rioxarray打开数据集，命名为ds_netcdf
# 创建待匹配的波段名称列表
expected_keys = ['max_apples', 'max_oranges']
# 获取数据集中的波段名称
ds_keys = list(ds_netcdf.keys())
# 求两个列表的交集，得到共同波段名称
common_key = list(set(ds_keys) & set(expected_keys))  # 注：原代码中max_keys应为expected_keys，此处修正笔误

出错的代码示例：

# 获取栅格的最大波段数
bandcount = up_sampled.common_key[0].shape[0]

解决方案 #

问题核心是点语法的使用限制：

• 点语法（up_sampled.band_name）仅适用于直接写死的波段名称，此时band_name是Dataset实际存在的属性/波段名
• common_key[0]是存储波段名字符串的变量，必须用方括号索引器实现动态访问：

# 正确写法：用方括号传入变量中的波段名称
bandcount = up_sampled[common_key[0]].shape[0]

如果要遍历所有共同波段，可使用循环：

for band_name in common_key:
    bandcount = up_sampled[band_name].shape[0]
    # 后续处理逻辑

另外注意原匹配代码中的笔误：set(max_keys)应为set(expected_keys)，否则会因未定义max_keys报错。

内容的提问来源于stack exchange，提问作者anakaine