2.Matplotlib设置座标轴详解

1 设置座标轴

Matplotlib中设置座标轴的样式很有必要的，这样绘制的图才能更加清楚，比如横、纵轴的标签、刻度等的设置。

1.1 设置x,y的范围，描述x,y的标签

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-3,3,100)
y1 = 2*x +1
y2 = x**2

# xy范围
plt.xlim(-1,2)
plt.ylim(-2,3)

# xy描述
plt.xlabel('I am X')
plt.ylabel('I am Y')

plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
plt.show()

1.2 x,y轴刻度的自定义

new_ticks = np.linspace(-2,2,11)
print(new_ticks)

[-2.  -1.6 -1.2 -0.8 -0.4  0.   0.4  0.8  1.2  1.6  2. ]

x = np.linspace(-3,3,100)
y1 = 2*x +1
y2 = x**2

# xy范围
plt.xlim(-1,2)
plt.ylim(-2,3)

# xy描述
plt.xlabel('I am X')
plt.ylabel('I am Y')

plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')

# 修改x的间隔，修改y为文字描述
plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
           ['level1','level2','level3','level4','level5'])
plt.show()

1.3 设置2D座标系

有时候我们需要绘制数学中2D的座标系，这时候就不是上边两小节的图，但是在上图的基础上进行设置的，比如去掉上边和右边的边框。

import matplotlib.pyplot as plt
import numpy as np

new_ticks = np.linspace(-2,2,11)
print(new_ticks)

[-2.  -1.6 -1.2 -0.8 -0.4  0.   0.4  0.8  1.2  1.6  2. ]

x = np.linspace(-3,3,100)
y1 = 2*x +1
y2 = x**2

# xy范围
plt.xlim(-1,2)
plt.ylim(-2,3)

# xy描述
plt.xlabel('I am X')
plt.ylabel('I am Y')

plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')

plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
           ['level1','level2','level3','level4','level5'])

# gca  get current axis
ax = plt.gca()
# 把右边和上边的边框去掉
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
# 把x轴的刻度设置为‘bottom'
# 把y轴的刻度设置为 ’left‘
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')
# 设置bottom对应到0点
# 设置left对应到0点
ax.spines['bottom'].set_position(('data',0))
ax.spines['left'].set_position(('data',0))

plt.show()