3d plot in Python - 條件 function 中缺少最大值

Question

我正在嘗試獲取 3d plot，其中我的 function 是有條件的。 但是，在這種情況下，最大值 - 1 丟失了。 如果我有一個沒有條件的 function，則 1 在圖表上：

import numpy as np
import matplotlib.pyplot as plt

from mpl_toolkits import mplot3d
x=np.linspace(0,1,100)
y = np.linspace(0,1,100)


def I_LK(x,y):
    v=1-x+y
    return np.where(v<=1, v, 1)
    #return 1-x+x*y # here there is no problem
    
X,Y = np.meshgrid(x,y)
Z=I_LK(X,Y)

fig = plt.figure()
ax = plt.axes(projection='3d')
ax.contour3D(X, Y, Z, 100, cmap='viridis')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
ax.set_zlim(0,1.1)

如何在圖表上獲得值“1”？

Answer 1

萬一有人找，正確的方法如下：

import numpy as np
import matplotlib.pyplot as plt

from mpl_toolkits import mplot3d
x=np.linspace(0,1,100)
y = np.linspace(0,1,100)


def I_LK(x,y):
    v=1-x+y
    return np.where(v<=1, v, 1)
    #return 1-x+x*y # here there is no problem
    
X,Y = np.meshgrid(x,y)
Z=I_LK(X,Y)

fig = plt.figure()
ax = plt.axes(projection='3d')
ax.plot_surface(X, Y, Z, cmap='viridis')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
ax.set_zlim(0,1.1)

plot_surface不是contour3D