數據可視化——matplotlib(2)

導入相關模塊

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

圖表設置

添加X、Y軸標簽以及圖標標題

a = (1,3,2,5,4)
b = (2,5,3,4,1)

plt.plot(a)
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Test')
plt.show()

添加圖例

簡單的圖例僅需要在畫圖函數中添加一個label標簽就行，在繪制完后調用plt.legend()函數即可，如果需要使用更復雜的圖例顯示，可以查看官網文檔，這里不做細說。

plt.plot(a, label='a')
plt.plot(b, label='b')
plt.legend()
plt.show()

坐標設置

設置坐標軸顯示范圍

坐標軸顯示范圍設置可以使用plt.xlim()和plt.ylim()函數實現。

plt.plot(a, label='a')
plt.xlim(xmin=0,xmax=10)
plt.ylim(ymin=0,ymax=10)
plt.show()

設置刻度

刻度的設置可以使用plt.xticks()和plt.yticks()來設置，需要傳入的是原刻度以及對應刻度，比如:plt.xticks(x, ticks)，這樣就可以將X軸顯示為想要顯示的刻度了。刻度也可以設置為傾斜的，只需要添加一個rotation參數即可。

plt.plot(a)
ticks = 'abcde'
plt.xticks(range(5),ticks,rotation=30)
plt.show()

數據標簽

plt中添加數據標簽可以通過plt.text()方法。具體用法是：plt.text(x,y,label)

plt.plot(a)
for i in range(len(a)):
    plt.text(i,a[i], a[i])
plt.show()

當然，plt.text()還可以設置其它的參數，比如：

• ha:horizontalalignment（水平對齊）
• va:verticalalignment（垂直對齊）
• fs:文字大小
  這些就不做說明了。
  至此，一些簡單的設置都已經實現了。