Python求梯度的方法是什么

55次阅读
没有评论

共计 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 网 – 提供最优质的资源集合!

正文完
 
丸趣
版权声明:本站原创文章,由 丸趣 2024-03-19发表,共计365字。
转载说明:除特殊说明外本站除技术相关以外文章皆由网络搜集发布,转载请注明出处。
评论(没有评论)