python matplotlib调整图的坐标轴位置

Posted 2021-05-09

"""
调整图的坐标轴位置

实际上每个图都带有4条坐标轴, 我们可以通过将其中两条颜色设置为None来达到隐藏的目的,
然后将另外两条移至图像指定坐标点的位置即可
"""

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-np.pi, np.pi, 500, endpoint=True)
y = np.sin(x)

plt.plot(x, y)

ax = plt.gca()

# hide two spines
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')

# move bottom and left spine to 0,0
# 因为本题图像的(0,0)点恰好在图中心, 所以效果就是将坐标轴移动到了图中心
ax.spines['bottom'].set_position(('data',0))
ax.spines['left'].set_position(('data',0))

# 指定将刻度和哪条坐标轴绑定, 通常绑定的是左边和底部的坐标轴
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

plt.show()