python - Adjusting gridlines on a 3D Matplotlib figure

Question

I'm getting ready for a presentation and I have some example figures of 3D matplotlib figures. However, the gridlines are too light to see on the projected images.

I tried using the grid-method that works for 2D figures:

points = (5*np.random.randn(3, 50)+np.tile(np.arange(1,51), (3, 1))).transpose()
fig = plt.figure(figsize = (10,10))
ax = fig.add_subplot(111, projection='3d') 
ax.scatter(points[:,0], points[:,1], points[:,2])
ax.view_init(elev=0., azim=0)
ax.set_ylim([0, 60])
ax.set_zlim([0, 60])
ax.set_xlim([0, 60])
ax.set_zlabel('Cytokine')
ax.set_ylabel('Parameter')
ax.grid(linewidth=20)

But that doesn't seem to work for 3D figures. Any suggestions?

Answer 1

如果您不介意让所有线条变粗，那么您可以调整默认的 rc 设置。

给定一个图形，例如：

我们可以添加：

import matplotlib as mpl
mpl.rcParams['lines.linewidth'] = 2

要增加所有行的默认线宽，结果如下：

或者，如果您觉得这看起来很难看，您可以使用：

ax.w_xaxis.gridlines.set_lw(3.0)
ax.w_yaxis.gridlines.set_lw(3.0)
ax.w_zaxis.gridlines.set_lw(3.0)

将每个轴的线宽调整为 3.0，产生：

为了更新颜色，使网格线真正弹出，您可以添加：

ax.w_xaxis._axinfo.update({'grid' : {'color': (0, 0, 0, 1)}})
ax.w_yaxis._axinfo.update({'grid' : {'color': (0, 0, 0, 1)}})
ax.w_zaxis._axinfo.update({'grid' : {'color': (0, 0, 0, 1)}})

产生：

这些方法非常老套，但据我所知，没有更简单的方法可以实现这些结果！！希望这可以帮助; 如果您需要任何进一步的帮助，请告诉我！

Answer 2

不幸的是，这似乎没有暴露出来。查看源代码，关键的内部变量是 call_AXINFO我们可以通过仔细的子类化来覆盖它。

创建图形后添加此代码，并使用 dict 对其进行样式设置custom_AXINFO：

from mpl_toolkits.mplot3d import Axes3D
import mpl_toolkits.mplot3d.axis3d as axis3d

# New axis settings
custom_AXINFO = {
    'x': {'i': 0, 'tickdir': 1, 'juggled': (1, 0, 2),
          'color': (0.00, 0.00, 0.25, .75)},
    'y': {'i': 1, 'tickdir': 0, 'juggled': (0, 1, 2),
          'color': (0.20, 0.90, 0.90, 0.25)},
    'z': {'i': 2, 'tickdir': 0, 'juggled': (0, 2, 1),
          'color': (0.925, 0.125, 0.90, 0.25)},}

class custom_XAxis(axis3d.Axis):
    _AXINFO = custom_AXINFO

class custom_YAxis(axis3d.Axis):
    _AXINFO = custom_AXINFO

class custom_ZAxis(axis3d.Axis):
    _AXINFO = custom_AXINFO

class custom_Axes3D(Axes3D):
    def _init_axis(self):
        '''Init 3D axes; overrides creation of regular X/Y axes'''
        self.w_xaxis = custom_XAxis('x', self.xy_viewLim.intervalx,
                                    self.xy_dataLim.intervalx, self)
        self.xaxis = self.w_xaxis
        self.w_yaxis = custom_YAxis('y', self.xy_viewLim.intervaly,
                            self.xy_dataLim.intervaly, self)
        self.yaxis = self.w_yaxis
        self.w_zaxis = custom_ZAxis('z', self.zz_viewLim.intervalx,
                            self.zz_dataLim.intervalx, self)
        self.zaxis = self.w_zaxis

        for ax in self.xaxis, self.yaxis, self.zaxis:
            ax.init3d()

# The rest of your code below, note the call to our new custom_Axes3D

points = (5*np.random.randn(3, 50)+np.tile(np.arange(1,51), (3, 1))).transpose()
fig = plt.figure(figsize = (10,10)) 
ax = custom_Axes3D(fig)

这是最糟糕的猴子补丁，不应该依赖于为以后的版本工作。

修复 facecolors 比网格线更容易，因为这需要覆盖其中一种__init__方法，尽管可以通过更多工作来完成。

将它暴露给最终用户似乎并不难，因此我可以想象这可能会在以后的版本中得到修复。

Answer 3

如果要使网格的背景变亮，可以使用如下 Axes3DSubplot 对象将窗格颜色设置为更亮（例如：白色）。

ax.w_xaxis.pane.set_color('w');
ax.w_yaxis.pane.set_color('w');
ax.w_zaxis.pane.set_color('w');

或者为了进一步突出网格线，可以更新绘图的网格颜色参数。

plt.rcParams['grid.color'] = "black"

Answer 4

只需使用它来更改线宽：

plt.rcParams['grid.linewidth'] = 2

绘制图形的完整脚本：

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

points = (5*np.random.randn(3, 50)+np.tile(np.arange(1,51), (3, 1))).transpose()
fig = plt.figure(figsize = (10,10))
ax = fig.add_subplot(111, projection='3d') 
ax.scatter(points[:,0], points[:,1], points[:,2])
#ax.view_init(elev=0., azim=0)
ax.set_ylim([0, 60])
ax.set_zlim([0, 60])
ax.set_xlim([0, 60])
ax.set_zlabel('Cytokine')
ax.set_ylabel('Parameter')
plt.rcParams['grid.linewidth'] = 4   # change linwidth
plt.rcParams['grid.color'] = "black" # change color