Изменение толщины линии сетки на трехмерном графике поверхности в Python Matplotlib
Я пытаюсь изменить толщину и прозрачность линий, которые составляют сетку на фоне графика поверхности, как этот пример с сайта Matplotlib:
Вот исходный код:
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import numpy as np
fig = plt.figure()
ax = fig.gca(projection='3d')
# Make data.
X = np.arange(-5, 5, 0.25)
Y = np.arange(-5, 5, 0.25)
X, Y = np.meshgrid(X, Y)
R = np.sqrt(X**2 + Y**2)
Z = np.sin(R)
# Plot the surface.
surf = ax.plot_surface(X, Y, Z, cmap=cm.coolwarm,
linewidth=0, antialiased=False)
# Customize the z axis.
ax.set_zlim(-1.01, 1.01)
ax.zaxis.set_major_locator(LinearLocator(10))
ax.zaxis.set_major_formatter(FormatStrFormatter('%.02f'))
# Add a color bar which maps values to colors.
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()
Я пробовал звонить ax.grid(linewidth=x)
но это, кажется, не имеет значения. Есть ли другой способ изменить толщину?
1 ответ
Решение
Способ установки параметров сетки в mplot3d - обновить _axinfo
словарь соответствующей оси.
Чтобы установить ширину линии сетки в направлении y, используйте, например,
ax.yaxis._axinfo["grid"]['linewidth'] = 3.
Вот общий пример:
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.set_xlabel("x"); ax.set_ylabel("y"); ax.set_zlabel("z")
print ax.xaxis._axinfo
ax.xaxis._axinfo["grid"].update({"linewidth":1, "color" : "green"})
ax.yaxis._axinfo["grid"]['linewidth'] = 3.
ax.zaxis._axinfo["grid"]['color'] = "#ee0009"
ax.zaxis._axinfo["grid"]['linestyle'] = ":"
plt.show()