plt.gca()坐标轴移动

我们可以看到绘制出来的图有四个边框，我们通过gca()对坐标轴进行一些简单处理，代码如下。

import torch
import torch.nn.functional as F
import matplotlib.pyplot as plt
import numpy as np
x = torch.linspace(-10,10,60)
y = torch.sigmoid(x)
ax = plt.gca()
ax.spines['top'].set_color('none') #将最上方的边框颜色置为none
ax.spines['right'].set_color('none') #将右边的边框颜色置为none
ax.xaxis.set_ticks_position('bottom') #要移动底部x轴，所以先要锁定x轴
ax.spines['bottom'].set_position(('data',0)) # 'data'表示按数值挪动，其后数字代表挪动到Y轴的刻度值
ax.yaxis.set_ticks_position('left') #同上
ax.spines['left'].set_position(('data',0)) #同上
plt.plot(x.numpy(),y.numpy())
plt.show()

#  要挪动底部的X轴，所以先目光锁定底部！