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()