以下是按指定的月份掩盖NumPy数组的解决方法的示例代码:
import numpy as np
# 创建示例数组
arr = np.array([
[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16]
])
# 指定要掩盖的月份
month_to_mask = 2
# 创建月份索引数组
months = np.array([
[1, 1, 1, 1],
[2, 2, 2, 2],
[3, 3, 3, 3],
[4, 4, 4, 4]
])
# 使用掩码数组来掩盖指定月份
masked_arr = np.ma.masked_where(months == month_to_mask, arr)
print("原始数组:")
print(arr)
print("\n掩盖指定月份后的数组:")
print(masked_arr)
输出:
原始数组:
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]
[13 14 15 16]]
掩盖指定月份后的数组:
[[ 1 2 3 4]
[-- -- -- --]
[ 9 10 11 12]
[13 14 15 16]]
在上面的示例中,我们首先创建了一个示例数组arr
,它是一个4x4的NumPy数组。然后,我们指定要掩盖的月份为2。接下来,我们创建了一个与arr
具有相同形状的月份索引数组months
,其中每个元素表示对应位置的月份。最后,我们使用np.ma.masked_where()
函数和掩码数组来创建一个掩盖了指定月份的新数组masked_arr
。最后,我们打印原始数组和掩盖指定月份后的数组。
上一篇:按指定的键将列表中的字典展平
下一篇:按指定格式创建数据摘要