在 NumPy 中沿深度方向（第三軸）依次堆疊掩碼陣列

要在深度方向（沿第三軸）依次堆疊掩碼陣列，請在 Python NumPy 中使用 **ma.dstack()** 方法。這等效於在將形狀為 (M,N) 的二維陣列重塑為 (M,N,1) 以及將形狀為 (N,) 的一維陣列重塑為 (1,N,1) 之後，沿第三軸進行連線。重建由 dsplit 分割的陣列。

對於最多 3 維的陣列，此函式最有意義。例如，對於具有高度（第一軸）、寬度（第二軸）和 r/g/b 通道（第三軸）的畫素資料。函式 concatenate、stack 和 block 提供更通用的堆疊和連線操作。

引數是必須沿除第三軸之外的所有軸具有相同形狀的陣列。一維或二維陣列必須具有相同的形狀。該函式返回由堆疊給定陣列形成的陣列，將至少為 3 維。

步驟

首先，匯入所需的庫 -

import numpy as np
import numpy.ma as ma

建立陣列 1，一個使用 numpy.arange() 方法的包含整數元素的 3x3 陣列 -

arr1 = np.arange(9).reshape((3,3))
print("Array1...
", arr1)
print("
Array type...
", arr1.dtype)

建立掩碼陣列 1 -

arr1 = ma.array(arr1)

掩碼陣列 1 -

arr1[0, 1] = ma.masked
arr1[1, 1] = ma.masked

顯示掩碼陣列 1 -

print("
Masked Array1...
",arr1)

建立陣列 2，另一個使用 numpy.arange() 方法的包含整數元素的 3x3 陣列 -

arr2 = np.arange(9).reshape((3,3))
print("
Array2...
", arr2)
print("
Array type...
", arr2.dtype)

建立掩碼陣列 2 -

arr2 = ma.array(arr2)

掩碼陣列 2 -

arr2[2, 1] = ma.masked
arr2[2, 2] = ma.masked

顯示掩碼陣列 2 -

print("
Masked Array2...
",arr2)

要在深度方向（沿第三軸）依次堆疊掩碼陣列，請使用 ma.dstack() 方法 -

print("
Result of stacking arrays depth wise...
",ma.dstack((arr1, arr2)))

示例

# Python ma.MaskedArray - Stack masked arrays in sequence depth wise (along third axis)

import numpy as np
import numpy.ma as ma

# Array 1
# Creating a 3x3 array with int elements using the numpy.arange() method
arr1 = np.arange(9).reshape((3,3))
print("Array1...
", arr1)
print("
Array type...
", arr1.dtype)

# Get the dimensions of the Array
print("
Array Dimensions...
",arr1.ndim)

# Get the shape of the Array
print("
Our Array Shape...
",arr1.shape)

# Get the number of elements of the Array
print("
Elements in the Array...
",arr1.size)

# Create a masked array
arr1 = ma.array(arr1)

# Mask Array1
arr1[0, 1] = ma.masked
arr1[1, 1] = ma.masked

# Display Masked Array 1
print("
Masked Array1...
",arr1)

# Array 2
# Creating another 3x3 array with int elements using the numpy.arange() method
arr2 = np.arange(9).reshape((3,3))
print("
Array2...
", arr2)
print("
Array type...
", arr2.dtype)

# Get the dimensions of the Array
print("
Array Dimensions...
",arr2.ndim)

# Get the shape of the Array
print("
Our Array Shape...
",arr2.shape)

# Get the number of elements of the Array
print("
Elements in the Array...
",arr2.size)

# Create a masked array
arr2 = ma.array(arr2)

# Mask Array2
arr2[2, 1] = ma.masked
arr2[2, 2] = ma.masked

# Display Masked Array 2
print("
Masked Array2...
",arr2)

# To stack masked arrays in sequence depth wise (along third axis), use the ma.dstack() method in Python Numpy
print("
Result of stacking arrays depth wise...
",ma.dstack((arr1, arr2)))

輸出

Array1...
[[0 1 2]
[3 4 5]
[6 7 8]]

Array type...
int64

Array Dimensions...
2

Our Array Shape...
(3, 3)

Elements in the Array...
9

Masked Array1...
[[0 -- 2]
[3 -- 5]
[6 7 8]]

Array2...
[[0 1 2]
[3 4 5]
[6 7 8]]

Array type...
int64

Array Dimensions...
2

Our Array Shape...
(3, 3)

Elements in the Array...
9

Masked Array2...
[[0 1 2]
[3 4 5]
[6 -- --]]

Result of stacking arrays depth wise...
[[[0 0]
[-- 1]
[2 2]]
[[3 3]
[-- 4]
[5 5]]
[[6 6]
[7 --]
[8 --]]]

更新於: 2022年2月3日

310 次瀏覽

啟動您的 職業生涯

透過完成課程獲得認證

開始學習
廣告