在 Numpy 中返回数组的上三角,并将上方对角线元素设置为零

numpyserver side programmingprogramming更新于 2025/9/24 23:07:17

要返回数组的上三角,请使用 Python Numpy 中的 numpy.triu() 方法 −

  • 第一个参数是输入数组
  • 第二个参数是"k",即其上方元素为零的对角线。

其中,k = 0(默认值)为主对角线,k < 0 为主对角线下方,k > 0 为主对角线上方。该函数返回一个数组的副本,其中第 k 条对角线下方元素为零。对于 ndim 超过 2 的数组,triu 将应用于最后两个轴。

步骤

首先,导入所需的库 −

import numpy as np

创建二维数组 −

arr = np.array([[36, 36, 78, 88], [92, 81, 98, 45], [22, 67, 54, 69 ], [69, 80, 80, 99]])

显示我们的数组 −

print("数组...
",arr)

获取数据类型 −

print("
数组数据类型...
",arr.dtype)

获取数组的维度 −

print("
数组维度...
",arr.ndim)

获取数组的形状 −

print("
我们的数组形状...
",arr.shape)

获取数组元素的数量 −

print("
数组中的元素...
",arr.size)

要返回数组的上三角,请使用 Python Numpy 中的 numpy.triu() 方法。第二个参数是"k",即元素位于其上方的对角线。其中,k = 0(默认值)为主对角线,k < 0 为下方,k > 0 为上方。数组 −

print("
结果...
",np.triu(arr, k = 0))

示例

import numpy as np

# 创建二维数组
arr = np.array([[36, 36, 78, 88], [92, 81, 98, 45], [22, 67, 54, 69], [69, 80, 80, 99]])

# 显示数组
print("数组...
",arr) # 获取数据类型 print("
数组数据类型...
",arr.dtype) # 获取数组的维度 print("
数组维度...
",arr.ndim) # 获取数组的形状 print("
我们的数组形状...
",arr.shape) # 获取数组元素的数量 print("
数组中的元素...
",arr.size) # 要返回数组的上三角,请使用 Python Numpy 中的 numpy.triu() 方法。 # 第一个参数是输入数组。 # 第二个参数是"k",即元素在 k 上方的对角线。 # k = 0(默认值)为主对角线,k < 0 为下方,k > 0 为上方。.' array print("
结果...
",np.triu(arr, k = 0))

输出

数组...
[[36 36 78 88]
[92 81 98 45]
[22 67 54 69]
[69 80 80 99]]

数组数据类型...
int64

数组维度...
2

我们的数组形状...
(4, 4)

数组中的元素...
16

相关文章