如何以变量名作为文件名保存NumPy掩码数组至NC文件？

核心问题解决：获取变量名对应的字符串 #

Python没有直接获取变量名的内置方法，但可以通过反向查找变量字典的方式实现，以下是两种可靠的实现方式：

1. 基础版（当前作用域内使用）

def get_var_name(var):
    # 遍历当前局部变量，匹配对应变量的名字
    for name, value in locals().items():
        if value is var:
            return name
    return None

2. 函数内通用版（跨作用域获取）

如果处理逻辑封装在函数里，推荐用inspect模块获取调用者的变量字典，兼容性更强：

import inspect

def get_var_name(var):
    # 获取调用者的局部变量集合
    callers_vars = inspect.currentframe().f_back.f_locals.items()
    # 返回匹配变量的名字（确保变量名唯一，否则返回第一个匹配项）
    return next(name for name, value in callers_vars if value is var)

完整代码示例（单变量保存） #

结合你的需求，完整实现从获取变量名到保存为.nc文件的流程，同时保留掩码信息：

import numpy as np
import numpy.ma as ma
from netCDF4 import Dataset
import os
import inspect

def get_var_name(var):
    callers_vars = inspect.currentframe().f_back.f_locals.items()
    return next(name for name, value in callers_vars if value is var)

# 示例掩码数组
arr = np.array([[5,9,2],[4,3,9],[6,3,1]])
var = ma.masked_array(arr, mask=False)

# 创建保存目录（不存在则自动生成）
os.makedirs('Saved_files', exist_ok=True)

# 生成目标文件名
var_name = get_var_name(var)
filename = f'Saved_files/{var_name}.nc'

# 保存为NetCDF文件，同时处理掩码
with Dataset(filename, mode='w', format='NETCDF4') as ncfile:
    # 创建维度（y对应行，x对应列）
    ncfile.createDimension('y', var.shape[0])
    ncfile.createDimension('x', var.shape[1])
    # 创建数据变量，用fill_value对应掩码的缺失值
    data_var = ncfile.createVariable(var_name, np.float64, ('y','x'), fill_value=var.fill_value)
    data_var[:,:] = var.data
    # 单独存储掩码（可选，方便后续直接读取掩码）
    mask_var = ncfile.createVariable(f'{var_name}_mask', bool, ('y','x'))
    mask_var[:,:] = var.mask

批量处理多个掩码数组 #

如果有大量数组需要处理，只需将数组放入列表循环处理即可：

# 示例多个掩码数组
var1 = ma.masked_array(np.array([[1,2],[3,4]]), mask=[[False, True],[False, False]])
var2 = ma.masked_array(np.array([[5,6,7],[8,9,10]]), mask=False)
var3 = ma.masked_array(np.array([[11,12],[13,14],[15,16]]), mask=[[True, False],[False, True],[True, False]])

# 批量处理列表
mask_arrays = [var1, var2, var3]

for arr in mask_arrays:
    try:
        arr_name = get_var_name(arr)
        file_path = f'Saved_files/{arr_name}.nc'
        with Dataset(file_path, mode='w', format='NETCDF4') as ncfile:
            ncfile.createDimension('y', arr.shape[0])
            ncfile.createDimension('x', arr.shape[1])
            data_var = ncfile.createVariable(arr_name, np.float64, ('y','x'), fill_value=arr.fill_value)
            data_var[:] = arr.data
            mask_var = ncfile.createVariable(f'{arr_name}_mask', bool, ('y','x'))
            mask_var[:] = arr.mask
        print(f"已保存：{file_path}")
    except StopIteration:
        print("无法获取变量名，跳过该数组")

注意事项 #

• 确保变量名唯一，否则get_var_name会返回第一个匹配的变量名；
• 使用with语句管理NetCDF文件，无需手动调用close()，避免资源泄漏；
• 存储掩码时，既可以通过fill_value让NetCDF自动识别缺失值，也可以单独存储掩码变量，便于后续灵活处理。

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