從 NumPy 中的掩碼陣列中返回指定的對角線
要返回指定的對角線,在 Python Numpy 中使用 ma.MaskedArray.diagonal 方法。掩碼陣列是標準 numpy.ndarray 和一個掩碼的組合。掩碼要麼是 nomask,表示關聯陣列的任何值都不無效,要麼是布林陣列,用來確定關聯陣列的每個元素的值是否有效。
步驟
首先,匯入所需的庫 −
import numpy as np import numpy.ma as ma
使用 numpy.array() 方法建立一個包含 int 元素的陣列 −
arr = np.array([[49, 85, 45], [67, 33, 59]]) print("Array...
", arr) print("
Array type...
", arr.dtype)
獲取陣列的維度 −
print("Array Dimensions...
",arr.ndim)
建立一個掩碼陣列並標記其中一些無效 −
maskArr = ma.masked_array(arr, mask =[[0, 0, 1], [ 0, 1, 0]]) print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype)
獲取掩碼陣列的維度 −
print("
Our Masked Array Dimensions...
",maskArr.ndim)
獲取掩碼陣列的形狀 −
print("
Our Masked Array Shape...
",maskArr.shape)
獲取掩碼陣列的元素數量 −
print("
Elements in the Masked Array...
",maskArr.size)
返回指定的對角線,在 Numpy 中使用 ma.MaskedArray.diagonal() 方法 −
print("
Result...
",maskArr.diagonal())
範例
import numpy as np import numpy.ma as ma # Create an array with int elements using the numpy.array() method arr = np.array([[55, 85, 59, 77], [67, 33, 39, 57], [29, 88, 51, 37], [56, 45, 99, 85]]) print("Array...
", arr) print("
Array type...
", arr.dtype) # Get the dimensions of the Array print("
Array Dimensions...
",arr.ndim) # Create a masked array and mask some of them as invalid maskArr = ma.masked_array(arr, mask =[[1, 1, 0, 0], [ 0, 0, 1, 0], [0, 0, 0, 1], [0, 1, 0, 0]]) print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype) # Get the dimensions of the Masked Array print("
Our Masked Array Dimensions...
",maskArr.ndim) # Get the shape of the Masked Array print("
Our Masked Array Shape...
",maskArr.shape) # Get the number of elements of the Masked Array print("
Elements in the Masked Array...
",maskArr.size) # To return specified diagonals, use the ma.MaskedArray.diagonal() method in Numpy print("
Result...
",maskArr.diagonal())
輸出
Array... [[55 85 59 77] [67 33 39 57] [29 88 51 37] [56 45 99 85]] Array type... int64 Array Dimensions... 2 Our Masked Array [[-- -- 59 77] [67 33 -- 57] [29 88 51 --] [56 -- 99 85]] Our Masked Array type... int64 Our Masked Array Dimensions... 2 Our Masked Array Shape... (4, 4) Elements in the Masked Array... 16 Result... [-- 33 51 85]
廣告