Изменение толщины линии сетки на трехмерном графике поверхности в 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()

введите описание изображения здесь

plt.rcParams['grid.linewidth'] = 3
Другие вопросы по тегам