1. 程式人生 > >python強大的繪圖模組matplotlib示例講解

python強大的繪圖模組matplotlib示例講解

Matplotlib 是 Python 的繪相簿。作為程式設計師,經常需要進行繪圖,在我自己的工作中,如果需要繪圖,一般都是將資料匯入到excel中,然後通過excel生成圖表,這樣操作起來還是比較繁瑣的,所以最近學習了一下Matplotlib模組,將該模組的常用的繪圖手段和大家分享一下,提高大家在工作中的效率;

在示例中,我們主要用到Matplotlib和Numpy這兩個模組來為大家演示Python強大的繪圖功能,相信大家通過我下面的10個示例,基本上可以滿足大家日常工作的需求,再次強調一下,只是簡單的用法,大家千萬不要想通過這篇部落格獲取到太高深的用法。

下面進入正題

1、繪製一條直線

程式碼如下,下面的程式碼大家應該都可以看懂吧

# 匯入常用的包
import numpy as np
import matplotlib.pyplot as plt
 
# 生成-1到1的資料,一共生成100個,也可以生成1到-1的資料,這些資料是平均分佈的
# 定義x軸的資料
x = np.linspace(-1,1,100)
 
# 定義y軸的資料
y = x * 2 + 100
plt.plot(x,y)
 
# 顯示影象
plt.show()

  

效果如下

 

2、建立一個畫布,同時設定該畫布的大小

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
x = np.linspace(-1,1,100)
 
y1 = x * 2 + 100
 
 
y2 = x ** 2
 
# 建立一個畫布
 
# figsize:設定畫布的大小
plt.figure(figsize=(2,2))
plt.plot(x,y1)
 
# 建立第二個畫布
plt.figure()
plt.plot(x,y2)
 
plt.show()

  

效果如下,會同時顯示兩張畫布

 

 3、在一張畫布中畫兩條線,同時可以設定線的顏色,寬度,和風格

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
x = np.linspace(-1,1,100)
 
y1 = x * 2 + 0.5
 
 
y2 = x ** 2
 
 
 
# color:表示設定線的顏色
# linewidth:表示設定線的寬度
# linestyle:表示設定線的風格
plt.figure(figsize=(2,2))
plt.plot(x,y1,color='r',linewidth=1.0,linestyle='--')
 
plt.plot(x,y2,color='b',linewidth=5.0,linestyle='-')
 
plt.show()
 
# 上面的效果就是2條曲線被放到一個畫布中

  

效果如下

 

 4、限制x軸,y軸的顯示範圍,為x軸和y軸新增描述,替換x軸和y軸的顯示資訊

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
# 設定座標軸
x = np.linspace(-3,3,100)
 
y1 = x * 2 + 0.5
 
y2 = x ** 2
 
 
plt.figure(figsize=(6,6))
plt.plot(x,y1,color='r',linewidth=1.0,linestyle='--')
 
plt.plot(x,y2,color='b',linewidth=5.0,linestyle='-')
 
# 限制x軸的顯示範圍
plt.xlim((-1,2))
 
# 限制y軸的顯示範圍
plt.ylim((-1,5))
 
# 給x軸加描述
plt.xlabel("xxxxxx")
# 給y軸加描述
plt.ylabel("yyyyyy")
 
 
 
# 替換一下橫座標的顯示
temp = np.linspace(-2,2,11)
plt.xticks(temp)
 
# 替換縱座標的標籤,用level0代替之前的-1
plt.yticks([-1,0,1,2,3,4,5],["level0","level1","level2","level3","level4","level5","level6"])
 
plt.show()

  

效果如下

 

 5、對邊框進行設定,調整x軸和y軸的位置

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
# 設定座標軸
x = np.linspace(-3,3,100)
 
y1 = x * 2 + 0.5
 
y2 = x ** 2
 
 
plt.figure(figsize=(6,6))
plt.plot(x,y1,color='r',linewidth=1.0,linestyle='--')
 
plt.plot(x,y2,color='b',linewidth=5.0,linestyle='-')
 
# 限制x軸的顯示範圍
plt.xlim((-1,2))
 
# 限制y軸的顯示範圍
plt.ylim((-1,5))
 
# 給x軸加描述
plt.xlabel("xxxxxx")
# 給y軸加描述
plt.ylabel("yyyyyy")
 
 
 
# 替換一下橫座標的顯示
temp = np.linspace(-2,2,11)
plt.xticks(temp)
 
# 替換縱座標的標籤,用level0代替之前的-1
# plt.yticks([-1,0,1,2,3,4,5],["level0","level1","level2","level3","level4","level5","level6"])
 
# 獲取邊框
ax = plt.gca()
# 設定右邊框的顏色為紅色
ax.spines["right"].set_color("r")
 
# 去掉上邊框
ax.spines["top"].set_color(None)
 
# 把x軸的刻度設定為bottom
ax.xaxis.set_ticks_position("bottom")
# 把y軸的客戶設定為left
ax.yaxis.set_ticks_position("left")
 
# 設定x和y交匯的點,x軸是0,y是也是0,也就是x軸和y軸的都是0點交匯
ax.spines["bottom"].set_position(("data",0))
ax.spines["left"].set_position(("data",0))
 
plt.show()

  

效果如下

 

6、為畫布新增圖例

程式碼如下

#Auther Bob
#--*--conding:utf-8 --*--
import numpy as np

import matplotlib.pyplot as plt

# 設定圖例
x = np.linspace(-3, 3, 100)

y1 = x * 2 + 0.5

y2 = x ** 2

plt.figure(figsize=(6, 6))

# 首先要為兩條線分別取名,這裡的逗號必須要有
l1, = plt.plot(x, y1, color='r', linewidth=1.0, linestyle='--')

l2, = plt.plot(x, y2, color='b', linewidth=5.0, linestyle='-')

# handles控制圖例中要說明的線
# labels為兩條線分別取一個label
# loc控制圖例的顯示位置,一般用best,由程式碼為我們選擇最優的位置即可

plt.legend(handles=[l1, l2], labels=["test1", "test2"], loc='best')

# 限制x軸的顯示範圍
plt.xlim((-1, 2))

# 限制y軸的顯示範圍
plt.ylim((-1, 5))

# 給x軸加描述
plt.xlabel("xxxxxx")
# 給y軸加描述
plt.ylabel("yyyyyy")

# 替換一下橫座標的顯示
temp = np.linspace(-2, 2, 11)
plt.xticks(temp)

# 替換縱座標的標籤,用level0代替之前的-1
plt.yticks([-1, 0, 1, 2, 3, 4, 5], ["level0", "level1", "level2", "level3", "level4", "level5", "level6"])

# 為影象加一個圖例,用來對影象做說明



plt.show()

  

效果如下

 

 7、為影象新增描述

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
# 為影象做標註
x = np.linspace(-3,3,100)
 
y1 = x * 2
 
# y2 = x ** 2
 
 
plt.figure(figsize=(6,6))
 
 
plt.plot(x,y1,color='r',linewidth=1.0,linestyle='-')
 
 
 
 
 
# 給x軸加描述
plt.xlabel("xxxxxx")
# 給y軸加描述
plt.ylabel("yyyyyy")
 
# ======================================================
# 在x軸為x0,y軸為x0 * 2上畫一個點,這個點的顏色是紅色,大小為50,這個大小就是這個點顯示的大小
x0 = 0.5
y0 = x0 * 2
# scatter是畫點的方法
plt.scatter(x0,y0,color='g',s=50)
 
# 畫線
# 這條線是第一個點的座標為[x0,y0],第二個點的座標為[x0,-6],後面就是設定線的風格,線的顏色,線的寬度
 
plt.plot([x0,x0],[y0,-6],color='k',linestyle='--',linewidth=1.0)
 
 
 
# 畫箭頭和描述
 
 
# xy代表我們的點
# xytext程式碼我們描述的位置,基於當前的點,在x軸+30,在y軸-30
# r'$2*x={n}$是我們要顯示的文字資訊,格式必須要這樣
# textcoords表示作為起點
# fontsize表示設定字型大小
# arrowprops設定箭頭
# arrowstyle設定箭頭的樣式
# connectionstyle設定風格.2表示弧度
plt.annotate(r'$2*0.5={n}$'.format(n = y0),xy=(x0,y0),xytext=(+30,-30),textcoords='offset points',fontsize=10,arrowprops=dict(arrowstyle='->',connectionstyle='arc3,rad=.2'))
 
# 顯示文字描述,從x軸為-1,y軸為2開始顯示,$$中就是要顯示的字元,這裡如果要顯示空格,則需要轉義
# fontdict設定字型
plt.text(-1,2,r'$1\ 2\ 3\ 4$',fontdict={"size":16,"color":"r"})
 
# =========================================================
 
# 為影象加一個圖例,用來對影象做說明
 
plt.show()

  

效果如下

 

8、繪製散點圖

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
 
 
 
# 繪製散點圖
# plt.scatter(np.arange(1,10,1),np.arange(10,19,1))
 
# plt.scatter(np.linspace(-3,3,10),np.linspace(-3,3,10))
 
 
x = np.random.normal(1,10,500)
y = np.random.normal(1,10,500)
 
print(x)
# s設定點的大小
# c是顏色
# alpha是透明度
plt.scatter(x,y,s=50,c='b',alpha=0.5)
plt.show()

  

效果如下

 

9、繪製直方圖

程式碼如下

import numpy as np
 
import matplotlib.pyplot as plt
 
 
# 繪製直方圖
x = np.arange(10)
y = x ** 2 + 10
 
 
# facecolor設定柱體的顏色
# edgecolor設定邊框的顏色
 
plt.bar(x,y,facecolor='g',edgecolor='r')
 
# 繪製翻轉過來的直方圖
# plt.bar(x,-y)
 
#顯示文字
for x,y in zip(x,y):
    plt.text(x,y,"{f}".format(f=y),ha="center",va='bottom')
plt.show()

  

效果如下

 

10、一張畫布顯示多張影象

程式碼如下

#Auther Bob
#--*--conding:utf-8 --*--
import numpy as np
import matplotlib.pyplot as plt




# plt.figure()
# 有一個兩行兩列的單元格,這個位於第一個單元格
# plt.subplot(2,2,1)

# 畫一條【0,0】-----》【1,1】的直線
# plt.plot([0,1],[0,1])



# 有一個兩行兩列的單元格,這個位於第一個單元格
# plt.subplot(2,2,2)

# 畫一條【0,0】-----》【1,1】的直線
# plt.plot([0,1],[0,1])



# 有一個兩行兩列的單元格,這個位於第一個單元格
# plt.subplot(2,2,3)

# 畫一條【0,0】-----》【1,1】的直線
# plt.plot([1,0],[0,1])
# plt.show()



# 上面的例子,每張圖他顯示的大小是一樣的,我們想顯示不同的大小該怎麼辦?

plt.figure()
# 有一個兩行三列的單元格,這個位於第一個單元格
plt.subplot(2,1,1)

# 畫一條【0,0】-----》【1,1】的直線
plt.plot([0,1],[0,1])



# 有一個兩行三列的單元格,這個位於第四個單元格,因為第一個單元格佔了3個位子,所以這裡就是第四個
plt.subplot(2,3,4)

# 畫一條【0,0】-----》【1,1】的直線
plt.plot([0,1],[0,1])



# 有一個兩行三列的單元格,這個位於第五個單元格
plt.subplot(2,3,5)

# 畫一條【0,0】-----》【1,1】的直線
plt.plot([1,0],[0,1])
plt.show()

  

效果如下

 

11、matplotlib模組中的顏色和線條風格,取自菜鳥教程

作為線性圖的替代,可以通過向 plot() 函式新增格式字串來顯示離散值。 可以使用以下格式化字元。

字元描述
'-' 實線樣式
'--' 短橫線樣式
'-.' 點劃線樣式
':' 虛線樣式
'.' 點標記
',' 畫素標記
'o' 圓標記
'v' 倒三角標記
'^' 正三角標記
'<' 左三角標記
'>' 右三角標記
'1' 下箭頭標記
'2' 上箭頭標記
'3' 左箭頭標記
'4' 右箭頭標記
's' 正方形標記
'p' 五邊形標記
'*' 星形標記
'h' 六邊形標記 1
'H' 六邊形標記 2
'+' 加號標記
'x' X 標記
'D' 菱形標記
'd' 窄菱形標記
'|' 豎直線標記
'_' 水平線標記

以下是顏色的縮寫:

字元顏色
'b' 藍色
'g' 綠色
'r' 紅色
'c' 青色
'm' 品紅色
'y' 黃色
'k' 黑色
'w' 白色