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軸，所以先目光鎖定底部！