matplotlib作图，刻度和背景颜色

本文的代码示例，主要是记录在用matplotlib作图时，如何设置坐标刻度和背景颜色。

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-2*np.pi, 2*np.pi, 1000)

fig = plt.figure('test', 
                 facecolor='lightyellow')
ax = fig.add_subplot(facecolor='lightgray')  
ax.spines['top'].set_color('none')
ax.spines['right'].set_color('none')
ax.spines['left'].set_position(('data', 0))
ax.spines['bottom'].set_position(('data', 0))
ax.plot(x, np.sin(x), linewidth=1, label=r'$y=\sin(x)$')
ax.plot(x, np.cos(x), linewidth=1, label=r'$y=\cos(x)$')
ax.plot(x, np.tan(x), linewidth=1, label=r'$y=\tan(x)$')
ax.set_ylim(-1.5, 1.5)
ax.set_xticks([-2*np.pi, 
               -1.5*np.pi, 
               -np.pi,
               -0.5*np.pi, 
               0,
               0.5*np.pi, 
               np.pi, 
               1.5*np.pi, 
               2*np.pi])
ax.set_xticklabels(
              [r'$-2\pi$', 
               r'$-\frac{3}{2}\pi$',
               r'$-\pi$',
               r'$-\frac{1}{2}\pi$',
               r'$0$',
               r'$\frac{1}{2}\pi$',
               r'$\pi$',
               r'$\frac{3}{2}\pi$',
               r'$2\pi$'])
               
ax.legend()
ax.set_xlabel('x')
ax.set_ylabel('y')

plt.show()

运行效果如下：

tan函数的垂直渐近线是自动生产的。

-- EOF --

相关文章

• 用matplotlib在3D空间画线
• 开始用matplotlib画函数图形
• matplotlib作图，设置figure和axes的title