在 Numpy 中按深度顺序(沿第三轴)堆叠掩码数组
numpyserver side programmingprogramming更新于 2025/9/27 1:37:17
要按深度顺序(沿第三轴)堆叠掩码数组，请使用 Python Numpy 中的 ma.dstack() 方法。这相当于将形状为 (M,N) 的二维数组重塑为 (M,N,1)，将形状为 (N,) 的一维数组重塑为 (1,N,1) 后，沿第三轴进行连接。重建数组后，除以 dsplit。
此函数最适合维度不超过 3 的数组。例如，对于具有高度(第一轴)、宽度(第二轴)和 r/g/b 通道(第三轴)的像素数据。函数 concatenate、stack 和 block 提供更通用的堆叠和连接操作。
参数是除第三轴外，其他所有轴必须具有相同形状的数组。一维或二维数组必须具有相同的形状。该函数返回由给定数组堆叠而成的数组，该数组至少为三维。
步骤
首先，导入所需的库 −
import numpy as np import numpy.ma as ma
使用 numpy.arange() 方法创建数组 1，一个包含 int 元素的 3x3 数组。 −
arr1 = np.arange(9).reshape((3,3))
print("Array1...
", arr1)
print("
数组类型...
", arr1.dtype)
创建带掩码的数组 1 −
arr1 = ma.array(arr1)
掩码数组 1 −
arr1[0, 1] = ma.masked arr1[1, 1] = ma.masked
显示掩码数组 1 −
print("
掩码数组 1...
",arr1)
使用 numpy.arange() 方法 − 创建数组 2，另一个包含 int 元素的 3x3 数组
arr2 = np.arange(9).reshape((3,3))
print("
数组 2...
", arr2)
print("
数组类型...
", arr2.dtype)
创建掩码数组 2 −
arr2 = ma.array(arr2)
掩码数组2 −
arr2[2, 1] = ma.masked arr2[2, 2] = ma.masked
显示掩码数组2 −
print("
掩码数组2...
",arr2)
要按深度顺序(沿第三轴)堆叠掩码数组，请使用 ma.dstack() 方法 −
print("
按深度堆叠数组的结果...
",ma.dstack((arr1, arr2)))
示例
# Python ma.MaskedArray - 按深度顺序堆叠掩码数组(沿第三轴)
import numpy as np
import numpy.ma as ma
# 数组 1
# 使用 numpy.arange() 方法创建一个包含 int 元素的 3x3 数组
arr1 = np.arange(9).reshape((3,3))
print("Array1...
", arr1)
print("
数组类型...
", arr1.dtype)
# 获取数组的维度
print("
数组维度...
",arr1.ndim)
# 获取数组的形状
print("
我们的数组形状...
",arr1.shape)
# 获取数组元素的数量
print("
数组中的元素...
",arr1.size)
# 创建一个掩码数组
arr1 = ma.array(arr1)
# 掩码数组 1
arr1[0, 1] = ma.masked
arr1[1, 1] = ma.masked
# 显示掩码数组 1
print("
掩码数组 1...
",arr1)
# 数组 2
# 使用 numpy.arange() 方法创建另一个包含 int 元素的 3x3 数组
arr2 = np.arange(9).reshape((3,3))
print("
数组 2...
", arr2)
print("
数组类型...
", arr2.dtype)
# 获取数组的维度
print("
数组维度...
",arr2.ndim)
# 获取数组的形状
print("
我们的数组形状...
",arr2.shape)
# 获取数组元素的数量
print("
数组中的元素...
",arr2.size)
# 创建一个掩码数组
arr2 = ma.array(arr2)
# 掩码数组 2
arr2[2, 1] = ma.masked
arr2[2, 2] = ma.masked
# 显示掩码数组 2
print("
掩码数组 2...
",arr2)
# 要按深度顺序堆叠掩码数组(沿第三轴)，请使用 Python Numpy 中的 ma.dstack() 方法
print("
按深度堆叠数组的结果...
",ma.dstack((arr1, arr2)))
输出
Array1... [[0 1 2] [3 4 5] [6 7 8]] 数组类型... int64 数组维度... 2 我们的数组形状... (3, 3) 数组中的元素... 9 掩码数组 1... [[0 -- 2] [3 -- 5] [6 7 8]] 数组 2... [[0 1 2] [3 4 5] [6 7 8]] 数组类型... int64 数组维度... 2 我们的数组形状... (3, 3) 数组中的元素... 9 掩码数组2... [[0 1 2] [3 4 5] [6 -- --]] 按深度堆叠数组的结果... [[[0 0] [-- 1] [2 2]] [[3 3] [-- 4] [5 5]] [[6 6] [7 --] [8 --]]]
