在 Numpy 中创建一个以扁平输入为对角线的二维数组
numpyserver side programmingprogramming
要创建一个以扁平输入为对角线的二维数组，请使用 Python Numpy 中的 numpy.diagflat() 方法。第一个参数是输入数据，它被扁平化并设置为输出的第 k 个对角线。第二个参数是要设置的对角线；默认值 0 对应于"主"对角线，正(负)k 表示主对角线上方(下方)对角线的数量。
NumPy 提供全面的数学函数、随机数生成器、线性代数例程、傅里叶变换等。它支持各种硬件和计算平台，并且与分布式、GPU 和稀疏数组库配合良好。
步骤
首先，导入所需的库 −
import numpy as np
使用 numpy.array() 方法创建二维数组 −
arr = np.array([[5, 10], [15, 20]])
显示我们的数组 −
print("数组...
",arr)
获取数据类型 −
print("
数组数据类型...
",arr.dtype)
获取数组的维度 −
print("
数组维度...
",arr.ndim)
获取数组的形状 −
print("
我们的数组形状...
",arr.shape)
获取数组元素的数量 −
print("
数组中的元素...
",arr.size)
要创建以扁平输入为对角线的二维数组，请使用 numpy.diagflat() 方法 −
print("
结果...
",np.diagflat(arr))
示例
import numpy as np
# 使用 numpy.array() 方法创建 2D 数组
arr = np.array([[5, 10], [15, 20]])
# 显示我们的数组
print("数组...
",arr)
# 获取数据类型
print("
数组数据类型...
",arr.dtype)
# 获取数组的维度
print("
数组维度...
",arr.ndim)
# 获取数组的形状
print("
我们的数组形状...
",arr.shape)
# 获取数组元素的数量
print("
元素数组...
",arr.size)
# 要创建一个以扁平输入为对角线的二维数组，请使用 Python Numpy 中的 numpy.diagflat() 方法
print("
结果...
",np.diagflat(arr))
输出
数组... [[ 5 10] [15 20]] 数组数据类型... int64 数组维度... 2 我们的数组形状... (2, 2) 数组中的元素... 4 结果... [[ 5 0 0 0] [ 0 10 0 0] [ 0 0 15 0] [ 0 0 0 20]]
