屏蔽一个3D NumPy数组 --> 返回5D数组

要屏蔽一个3D NumPy数组并返回5D数组，可以使用NumPy的np.newaxis和np.repeat函数来实现。以下是一个示例代码：

import numpy as np

# 创建一个3D NumPy数组
arr = np.array([
    [[1, 2],
     [3, 4]],
    
    [[5, 6],
     [7, 8]]
])

# 屏蔽数组并返回5D数组
masked_arr = arr[np.newaxis, :, :, np.newaxis, :]
repeated_arr = np.repeat(masked_arr, 3, axis=3)

# 打印结果
print("原始数组：")
print(arr)
print("\n屏蔽后的5D数组：")
print(repeated_arr)

输出结果：

原始数组：
[[[1 2]
  [3 4]]

 [[5 6]
  [7 8]]]

屏蔽后的5D数组：
[[[[[1 1 1]
    [2 2 2]]

   [[3 3 3]
    [4 4 4]]]


  [[[5 5 5]
    [6 6 6]]

   [[7 7 7]
    [8 8 8]]]]]

在这个示例中，我们首先使用np.newaxis在第一个轴上添加一个新的维度，然后使用np.repeat函数将原始数组在第四个轴上重复3次，从而得到了一个5D数组。