在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方向上的梯度值。