如何使用matplotlib绘制学生风格的图表？

Question

我目前在尝试使用matplotlib。一段时间以前，我使用Excel VBA代码生成了像所附图像一样的图像。

您会注意到，它不是以科学/研究风格呈现的，而是像一个学生用格纸制作的 - 带有三种不同的网格线样式。

是否有一种相当简单的方法来使用matplotlib实现这种效果？

- Geddes

Answer 1

是的，你可以使用坐标轴来完成这个任务。

import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator, FormatStrFormatter
import numpy as np

fig = plt.figure(1)
ax = fig.add_subplot(111)

# set up axis
ax.spines['left'].set_position('zero')
ax.spines['right'].set_color('none')
ax.spines['bottom'].set_position('zero')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

# draw curve
x = np.arange(-2.5,2.5,0.01)
line, = ax.plot(x, x**2)

#set bounds
ax.set_ybound(-1,7)

# create grid
#ax.xaxis.set_major_locator(MultipleLocator(1))
#ax.xaxis.set_minor_locator(MultipleLocator(0.2))
#ax.yaxis.set_major_locator(MultipleLocator(1))
#ax.yaxis.set_minor_locator(MultipleLocator(0.2))
#ax.xaxis.grid(True,'minor')
#ax.yaxis.grid(True,'minor')
#ax.xaxis.grid(True,'major',linewidth=2)
#ax.yaxis.grid(True,'major',linewidth=2)

#adjust grid on the 2s
#for idx,loc in enumerate(ax.xaxis.get_majorticklocs()):
    #if loc !=0 and loc % 2 == 0: ax.get_xgridlines()[idx].set_c('r')
#for idx,loc in enumerate(ax.yaxis.get_majorticklocs()):
    #if loc !=0 and loc % 2 == 0: ax.get_ygridlines()[idx].set_c('r')

## THIS IS THE EDIT
ax.xaxis.set_minor_locator(MultipleLocator(0.2))
ax.yaxis.set_minor_locator(MultipleLocator(0.2))
ax.xaxis.grid(True,'minor',linewidth=2)
ax.yaxis.grid(True,'minor',linewidth=2)

minor_grid_lines = [tick.gridline for tick in ax.xaxis.get_minor_ticks()]
for idx,loc in enumerate(ax.xaxis.get_minorticklocs()):
    if loc % 2.0 == 0: minor_grid_lines[idx].set_c('r' )
    elif loc % 1.0 == 0: minor_grid_lines[idx].set_c('g' )
    else: minor_grid_lines[idx].set_c( 'b' )

plt.show()

- Mark

---

我的最后一个问题是：有没有办法获得三种不同的网格线样式？我想要一个样式用于0.2，另一个用于1，还有一个用于*2（就像图纸上的）。这似乎很困难，因为Matplotlib似乎只强制使用主/次刻度线。再次感谢！ - Geddes

@ Geddes，请参见上面的编辑。我认为最简单的方法是根据它们的位置（即2的倍数）调整网格线。 - Mark

再次感谢你，马克。我现在得理解这些内容了——有些相当高级的东西！祝一切顺利。 - Geddes

Answer 2

这是上面被接受答案的修改版。 也许有人会发现这很有用。

import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator, FormatStrFormatter
import numpy as np
from matplotlib.ticker import FormatStrFormatter

_fontsize_legend = 10
_fontsize = 15

DP = 2

fig = plt.figure(figsize=(12, 12), dpi=100, facecolor='w', edgecolor='k')
##fig = plt.figure()
fig.canvas.draw()
ax = plt.gca()

# set up axis
ax.spines['left'].set_position('zero')
ax.spines['right'].set_color('none')
ax.spines['bottom'].set_position('zero')
ax.spines['top'].set_color('none')
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

# draw curve
x = np.arange(-2.5,2.5,0.01)
line, = ax.plot(x, x**2)

#set bounds
ax.set_ybound(-1,7)

## THIS IS THE EDIT
ax.xaxis.set_major_locator(MultipleLocator(1/4))
ax.yaxis.set_major_locator(MultipleLocator(1/4))
ax.xaxis.grid(True,'major',linewidth=2/DP,linestyle='-',color='#d7d7d7',zorder=0)
ax.yaxis.grid(True,'major',linewidth=2/DP,linestyle='-',color='#d7d7d7')

ax.xaxis.set_minor_locator(MultipleLocator( (1/4) / 5 ))
ax.yaxis.set_minor_locator(MultipleLocator( (1/4) / 5 ))
ax.xaxis.grid(True,'minor',linewidth=0.5/DP,linestyle='-',color='#d7d7d7')
ax.yaxis.grid(True,'minor',linewidth=0.5/DP,linestyle='-',color='#d7d7d7')

ax.set_axisbelow(True)
ax.set_aspect('equal')

##ax.axhline(linewidth=0)
##ax.axvline(linewidth=0)

ax.xaxis.set_major_formatter(FormatStrFormatter('%i'))
xticks = ax.xaxis.get_major_ticks()
for i,l in enumerate(xticks):
    if not (i - 1) % 4 == 0:
        xticks[i].label1.set_visible(False)
    else:
        xticks[i].label1.set_fontsize(_fontsize)

ax.yaxis.set_major_formatter(FormatStrFormatter('%i'))
yticks = ax.yaxis.get_major_ticks()
for i,l in enumerate(yticks):
    if not (i - 1) % 4 == 0:
        yticks[i].label1.set_visible(False)
    else:
        yticks[i].label1.set_fontsize(_fontsize)    

figManager = plt.get_current_fig_manager()
figManager.window.showMaximized()
plt.show()

- mrpetrkol

Answer 3

另一个想法 - 我也尝试过只使用次要网格线来完成所有操作（除此之外，这将有助于我的理解），但它没有正确枚举，毫无疑问是由于 get_minorticklocs 和 ax.get_xgridlines 的原因。抱歉，并提前致谢...

Geddes

import matplotlib.pyplot as plt 
from matplotlib.ticker import MultipleLocator, FormatStrFormatter 
import numpy as np 

fig = plt.figure(1) 
ax = fig.add_subplot(111) 

# set up axis 
ax.spines['left'].set_position('zero') 
ax.spines['right'].set_color('none') 
ax.spines['bottom'].set_position('zero') 
ax.spines['top'].set_color('none') 
ax.xaxis.set_ticks_position('bottom') 
ax.yaxis.set_ticks_position('left') 

# draw curve 
x = np.arange(-2.5,2.5,0.01) 
line, = ax.plot(x, x**2) 

#set bounds 
ax.set_ybound(-1,7) 

# create grid 
ax.xaxis.set_minor_locator(MultipleLocator(0.2)) 
ax.yaxis.set_minor_locator(MultipleLocator(0.2)) 
ax.xaxis.grid(True,'minor',linewidth=2) 
ax.yaxis.grid(True,'minor',linewidth=2) 

#adjust grid on the 2s 
for idx,loc in enumerate(ax.xaxis.get_minorticklocs()):
    if loc % 2 == 0: ax.get_xgridlines()[idx].set_color('r')
    if loc % 1 == 0: ax.get_xgridlines()[idx].set_color('g')
    if loc % 0.2 == 0: ax.get_xgridlines()[idx].set_color('b')

for idx,loc in enumerate(ax.yaxis.get_majorticklocs()): 
    if loc % 2 == 0: ax.get_ygridlines()[idx].set_c('b') 

plt.savefig('spines3.png',dpi=300)

- Geddes

---

@Geddes，ax.get_xgridlines()仅返回主要网格线。请参见上面我的答案的编辑。 - Mark

太棒了 - 马克，非常感谢你今天给我的所有帮助 - 我知道我是一个苛刻的顾客！祝一切顺利， 盖茨 - Geddes