我想使用 2 變量函數 Z(X,Y) 生成等值線圖。但是，我想施加一個條件，當 X 小于/大于某個值時更改函數。這樣的更改將允許我僅使用單行plt.contour（即我不想創建兩個單獨定義的函數，這會導致使用兩個單獨的繪圖命令行）。我繼續遇到（我認為的）真理/邏輯錯誤。我的猜測是 numpy 網格網格的某些方面不符合函數的條件“切換”。下面附上顯示該概念的簡短代碼以及完整的回溯錯誤。如果有任何不清楚的地方，或者我提供的內容不足以解釋我的問題，請隨時在下面發表評論。import numpy as npimport matplotlib.pyplot as pltX = np.linspace(0,50,100)Y = np.linspace(0,50,100)X, Y = np.meshgrid(X,Y)def z(x,y): if x < 20: return np.sin(x) + np.cos(y) else: return np.tan(x * y)Z = z(X,Y)plt.contourf(X, Y, Z)plt.xlabel('x')plt.ylabel('y')plt.colorbar()ValueError Traceback (most recent call last)<ipython-input-29-7e200be093e6> in <module> 16 17 ---> 18 Z = z(X,Y) 19 20 plt.figure(figsize=(8,6))<ipython-input-29-7e200be093e6> in z(x, y) 9 10 def z(x,y):---> 11 if x < 20: 12 return np.sin(x) + np.cos(y) 13 ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()```
3 回答
慕的地10843
TA貢獻1785條經驗 獲得超8個贊
您可以簡單地使用numpy.where().
你必須通過：
條件作為第一個參數
如果條件為真，則從中選擇值作為第二個參數
如果條件為假，則從中選擇值作為第三個參數
這樣，你的z()函數就變成了：
def?z(x,?y): ????return?np.where(x?<?20, ????????????????????np.sin(x)?+?np.cos(y), ????????????????????np.tan(x*y))
生成的情節：
守著一只汪
TA貢獻1872條經驗 獲得超4個贊
一切都很好，但你必須改變比較np.any(x<20)。這意味著如果 x 的任何元素大于 20。np.all如果您想在數組的每個元素都滿足條件時執行此操作，也可以使用 ,
import numpy as np
import matplotlib.pyplot as plt
X = np.linspace(0,50,100)
Y = np.linspace(0,50,100)
X, Y = np.meshgrid(X,Y)
def z(x,y):
if np.any(x < 20):
return np.sin(x) + np.cos(y)
else:
return np.tan(x * y)
Z = z(X,Y)
plt.contourf(X, Y, Z)
plt.xlabel('x')
plt.ylabel('y')
plt.colorbar()
郎朗坤
TA貢獻1921條經驗 獲得超9個贊
numpy 數組中的真實性測試是按元素進行的，
import numpy as np
X = np.array([1,2,3,4])
print(X<3)
輸出：[真真假假]
您可以使用.all()或.any()，具體取決于您是否希望所有或任何元素滿足條件
print((X<3).all())
輸出：假
添加回答
舉報
0/150
提交
取消