np.save()可复现示例正常但自定义数组报NotImplementedError，如何解决？

问题根源 #

你的variable_data本质是numpy.ma.MaskedArray类型，即便ma.is_masked()返回False（无有效掩码），它仍不是普通的numpy ndarray。np.save()在处理数组时会调用对象的tofile()方法，而MaskedArray的该方法目前未实现，因此触发NotImplementedError。

解决方案 #

根据是否需要保留掩码信息，选择以下两种方式：

1. 转换为普通ndarray后保存（无需掩码信息） #

如果确认不需要保留掩码相关数据，直接提取底层普通数组或转换类型即可：

# 方式一：提取MaskedArray的底层数据
np.save("delme2", variable_data.data)

# 方式二：强制转换为普通ndarray
normal_arr = np.asarray(variable_data)
np.save("delme2", normal_arr)

2. 使用掩码数组专用方法保存（需保留掩码信息） #

如果后续需要复用掩码结构（哪怕当前无掩码），使用numpy.ma模块提供的专用保存方法：

# 保存MaskedArray
np.ma.save("delme2_masked", variable_data)

# 读取时对应使用np.ma.load
loaded_masked_arr = np.ma.load("delme2_masked.npy")

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