Matplotlib 平滑 3D 表面数据

Question

我在使用 matplotlib 平滑 3D 表面的网格表示时遇到问题。下面，请看我的例子。如果可能的话，我很难弄清楚如何使情节看起来 nicer/smoother。感谢您提前抽出时间！

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.colors import LightSource
import numpy as np

X = [1,1,1,1,1,1,50,50,50,50,50,50]

Y = [3,5,7,8,9,10,3,5,7,8,9,10]

Z = [5.23,3.11,17.54,0.93,40.11,10.15,1.47,14.32,5.46,55.93,40.8,10.2]

x = np.reshape(X, (2, 6))
y = np.reshape(Y, (2, 6))
z = np.reshape(Z, (2, 6))
X, Y = np.meshgrid(x, y)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

ax.plot_surface(x, y, z)

ax.set_xlabel('Persistence Length')
ax.set_ylabel('Complexity')
ax.set_zlabel('Relative number of configurational states')

surf = ax.plot_surface(x, y, z, cmap=cm.coolwarm, 
                       linewidth=0, antialiased=False)
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()

Answer 1

要获得平滑 line/surface，您可以在曲面图上设置 antialiased=True。请注意，您正在绘制两个相同的表面：在下面的示例中，我删除了第一个。

要获得更平滑的网格，您可能需要在数据点之间进行插值。一种方法是使用 scipy.interpolate 模块中的 griddata。

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np
from scipy.interpolate import griddata

X = [1,1,1,1,1,1,50,50,50,50,50,50]

Y = [3,5,7,8,9,10,3,5,7,8,9,10]

Z = [5.23,3.11,17.54,0.93,40.11,10.15,1.47,14.32,5.46,55.93,40.8,10.2]

points = np.array([X, Y]).T

# create a grid of coordinates between the minimum and
# maximum of your X and Y. 50j indicates 50 discretization
# points between the minimum and maximum.
X_grid, Y_grid = np.mgrid[1:50:50j, 3:10:50j]
# interpolate your values on the grid defined above
Z_grid = griddata(points, Z, (X_grid, Y_grid), method='cubic')

fig = plt.figure(constrained_layout=True)
ax = fig.add_subplot(111, projection='3d')

ax.set_xlabel('Persistence Length')
ax.set_ylabel('Complexity')
ax.set_zlabel('Relative number of configurational states')

surf = ax.plot_surface(X_grid, Y_grid, Z_grid, cmap=cm.coolwarm, 
                       linewidth=0, antialiased=True)
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()

这里是左侧 antialiased=False 与右侧 antialiased=True 的示例：

Matplotlib 平滑 3D 表面数据

Matplotlib smoothing 3D surface data

python

matplotlib