如何解决numpy.ma.core.MaskError：无法修改被屏蔽元素的问题？

问题场景 #

将GitHub上的Python2代码迁移到Python3（Python3.7 + numpy1.24.1）时，运行代码触发以下报错：

Traceback (most recent call last):
  File "jsons2dssat.py", line 1600, in run
    sfileoutput = SOLFileOutput(sfile, efile, use_ptransfer = pfcn)
  File "jsons2dssat.py", line 1336, in init
    if isMaskedArray(vl): vl[vl.mask] = -99
  File "/conda/pSIMS/lib/python3.7/site-packages/numpy/ma/core.py", line 3348, in setitem
raise MaskError('Cannot alter the masked element.')
numpy.ma.core.MaskError: Cannot alter the masked element.

对应的代码片段是：

for var in soil_vars:
    v = f.variables[var]

    if 'profile' in v.dimensions and 'depth' in v.dimensions: # layer parameter
        for j in range(ndepths):
            vl = v[i, j, 0, 0]
            if isMaskedArray(vl): vl[vl.mask] = -99

报错原因 #

在Python2的旧版numpy中，可能允许直接修改单个masked标量的值，但在Python3的numpy（尤其是1.24.1版本）中，当vl是标量类型的masked值（而非数组）时，vl[vl.mask]的写法会触发错误——numpy禁止修改标量masked元素。

可行解决方案 #

方案1：用filled()直接生成填充后的值 #

这是最简洁的处理方式，直接将masked元素替换为指定值，返回新的非masked对象：

import numpy as np

for var in soil_vars:
    v = f.variables[var]

    if 'profile' in v.dimensions and 'depth' in v.dimensions: # layer parameter
        for j in range(ndepths):
            vl = v[i, j, 0, 0]
            if isinstance(vl, np.ma.MaskedArray):
                vl = vl.filled(-99)
            # 后续正常使用vl即可

filled(-99)会把所有masked元素替换为-99，如果是标量masked值，直接返回-99；如果是masked数组，返回填充后的普通数组。

方案2：区分标量和数组分别处理 #

如果需要保留原数组结构（仅修改masked元素），可以先判断是否为标量：

import numpy as np

for var in soil_vars:
    v = f.variables[var]

    if 'profile' in v.dimensions and 'depth' in v.dimensions: # layer parameter
        for j in range(ndepths):
            vl = v[i, j, 0, 0]
            if isinstance(vl, np.ma.MaskedArray):
                if vl.shape == ():  # 标量masked值
                    vl = -99 if vl.mask else vl
                else:  # masked数组
                    vl[vl.mask] = -99

方案3：读取时直接填充 #

可以在读取数据的一步就完成填充，省略后续判断：

vl = v[i, j, 0, 0].filled(-99)

这种方式不管是不是masked对象，都会返回填充后的值，无需额外判断。

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