共计 365 个字符,预计需要花费 1 分钟才能阅读完成。
在 Python 中,可以使用 NumPy 库中的 np.gradient()
函数来求梯度。该函数可以计算多维数组在各个维度上的梯度。示例如下:
import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 求数组 arr 在 x 和 y 方向上的梯度
grad_x, grad_y = np.gradient(arr)
print(" 梯度在 x 方向上:")
print(grad_x)
print("\n 梯度在 y 方向上:")
print(grad_y)
输出结果为:
梯度在 x 方向上:[[3. 3. 3.]
[3. 3. 3.]
[3. 3. 3.]]
梯度在 y 方向上:[[1. 1. 1.]
[1. 1. 1.]
[1. 1. 1.]]
可以看到,np.gradient()
函数返回了在 x 和 y 方向上的梯度值。
丸趣 TV 网 – 提供最优质的资源集合!
正文完