我正在嘗試繪制這樣的函數。但是，下面的代碼會導致錯誤。我認為原因是在 中返回了一個標量值norm ()，但是如何解決呢？圖像的標簽從左邊開始代表定義公式、搜索空間、最優解import numpy as npimport matplotlib.pyplot as pltx = np.arange(-5, 5, 0.05) y = np.arange(-5, 5, 0.05) X ,Y= np.meshgrid(x, y)print(X)c1 = -2 * np.ones((2,200,200))c2 = 4 * np.ones((2,200,200))print(np.linalg.norm(np.array([X,Y]) - c1))Z = (1 - 1 / (1 * np.linalg.norm(np.array([X,Y]) - c1) + 1)) + (1 - 1 / (2 * np.linalg.norm(np.array([X,Y]) - c2) + 1))plt.pcolormesh(X, Y, Z,cmap='hsv') plt.show()
1 回答
FFIVE
TA貢獻1797條經驗 獲得超6個贊
問題是您的 currentZ與您的X和 的維度不同Y。這可以通過打印 X、Y 和 Z 的形狀來驗證。原因是您沒有在方程式中提供axiswhile 計算norm，因此您得到了一個標量值。您可以參考官方文檔以獲取有關axis參數如何工作的更多信息。在您的情況下，由于您沒有為 指定任何值axis，因此它返回的是矩陣范數而不是向量范數
以下是您提供的解決方案，axis=0用于正確計算您X和Y
Z = (1 - 1 / (1 * np.linalg.norm(np.array([X,Y]) - c1, axis=0) + 1)) + (1 - 1 / (2 *np.linalg.norm(np.array([X,Y]) - c2, axis=0) + 1))
plt.pcolormesh(X, Y, Z,cmap='hsv')
添加回答
舉報
0/150
提交
取消