在 NumPy 中返回数组的下三角并置零主对角线
要返回数组的下三角,请在 Python NumPy 中使用 **numpy.tril()** 方法。第一个参数是输入数组。第二个参数是“k”,即要将元素置零的对角线上方。这里:
- k = 0(默认值)为主对角线,k < 0 在其下方,k > 0 在其上方。
- k = -1 的值也是为了将主对角线置零。
该函数返回一个数组的副本,其中高于第 k 个对角线的元素被置零。对于维度超过 2 的数组,tril 将应用于最后的两个轴。
步骤
首先,导入所需的库:
import numpy as np
创建一个二维数组:
arr = np.array([[36, 36, 78, 88], [92, 81, 98, 45], [22, 67, 54, 69 ], [69, 80, 80, 99]])
显示我们的数组:
print("Array...
",arr)
获取数据类型:
print("
Array datatype...
",arr.dtype)
获取数组的维度:
print("
Array Dimensions...
",arr.ndim)
获取数组的形状:
print("
Our Array Shape...
",arr.shape)
获取数组的元素数量:
print("
Elements in the Array...
",arr.size)
要返回数组的下三角,请使用 numpy.tril() 方法。第二个参数是“k”,即要将元素置零的对角线上方:
print("
Result...
",np.tril(arr, k = -1))
示例
import numpy as np # Create a 2d array arr = np.array([[36, 36, 78, 88], [92, 81, 98, 45], [22, 67, 54, 69], [69, 80, 80, 99]]) # Displaying our array print("Array...
",arr) # Get the datatype print("
Array datatype...
",arr.dtype) # Get the dimensions of the Array print("
Array Dimensions...
",arr.ndim) # Get the shape of the Array print("
Our Array Shape...
",arr.shape) # Get the number of elements of the Array print("
Elements in the Array...
",arr.size) # To return the lower triangle of an array, use the numpy.tril() method in Python Numpy # The 1st parameter is the input array # The 2nd parameter is the 'k' i.e. the diagonal above which to zero elements. # k = 0 (the default) is the main diagonal, k < 0 is below it and k > 0 is above.' array # The k = -1 value is to zero the main diagonal as well print("
Result...
",np.tril(arr, k = -1))
输出
Array... [[36 36 78 88] [92 81 98 45] [22 67 54 69] [69 80 80 99]] Array datatype... int64 Array Dimensions... 2 Our Array Shape... (4, 4) Elements in the Array... 16 Result... [[ 0 0 0 0] [92 0 0 0] [22 67 0 0] [69 80 80 0]]
广告