返回 NumPy 中掩碼陣列的對角線之和
要返回掩碼陣列元素的對角線之和,請在 NumPy 中使用 **ma.MaskedArray.trace()**。offset 引數是從主對角線偏移的對角線的偏移量。可以是正數也可以是負數。預設為 0。
axis 1 和 axis 2 是用作 2-D 子陣列的第一軸和第二軸的軸,從中應獲取對角線。預設為 a 的前兩個軸。dtype 確定返回陣列和累加器的資料型別,其中元素被求和。如果 dtype 的值為 None 且 a 為整數型別且精度小於預設整數精度,則使用預設整數精度。否則,精度與陣列的精度相同。
步驟
首先,匯入所需的庫 -
import numpy as np import numpy.ma as ma
使用 numpy.array() 方法建立包含 int 元素的陣列 -
arr = np.array([[55, 85, 68, 84], [67, 33, 39, 53], [29, 88, 51, 37], [56, 45, 99, 85]])
print("Array...
", arr)
print("
Array type...
", arr.dtype)獲取陣列的維度 -
print("
Array Dimensions...
",arr.ndim)
建立一個掩碼陣列並將其中一些標記為無效 -
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)獲取掩碼陣列的維度 -
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.trace() -
res = maskArr.trace()
print("
Result..
.", res)示例
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, 68, 84], [67, 33, 39, 53], [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 the sum along diagonals of the masked array elements, use the ma.MaskedArray.trace() in Numpy
res = maskArr.trace()
print("
Result..
.", res)輸出
Array... [[55 85 68 84] [67 33 39 53] [29 88 51 37] [56 45 99 85]] Array type... int64 Array Dimensions... 2 Our Masked Array [[-- -- 68 84] [67 33 -- 53] [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.. . 169.0
廣告
資料結構
網路
關係資料庫管理系統
作業系統
Java
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP