共計(jì) 365 個(gè)字符，預(yù)計(jì)需要花費(fèi) 1 分鐘才能閱讀完成。

在 Python 中，可以使用 NumPy 庫(kù)中的 np.gradient() 函數(shù)來(lái)求梯度。該函數(shù)可以計(jì)算多維數(shù)組在各個(gè)維度上的梯度。示例如下：

import numpy as np

# 創(chuàng)建一個(gè)二維數(shù)組
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 求數(shù)組 arr 在 x 和 y 方向上的梯度
grad_x, grad_y = np.gradient(arr)

print(" 梯度在 x 方向上：")
print(grad_x)

print("\n 梯度在 y 方向上：")
print(grad_y)

輸出結(jié)果為：

梯度在 x 方向上：[[3. 3. 3.]
 [3. 3. 3.]
 [3. 3. 3.]]

梯度在 y 方向上：[[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]

可以看到，np.gradient()函數(shù)返回了在 x 和 y 方向上的梯度值。

丸趣 TV 網(wǎng) – 提供最優(yōu)質(zhì)的資源集合！