本文約2800字,建議閱讀10分鐘
本文為你分享16個matplotlib繪圖實用小技巧。
標籤:數據分析
1. 添加標題-title
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlinex=np.arange(0,10)plt.title('這是一個示例標題')plt.plot(x,x*x)plt.show()/<code>
輸出:
2. 添加文字-text
設置座標和文字即可
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlinex=np.arange(-10,11,1)y=x*xplt.plot(x,y)plt.title('這是一個示例標題')# 添加文字plt.text(-2.5,30,'function y=x*x')plt.show()/<code>
輸出:
3. 添加註釋-annotate
- xy:為備註的座標點
- xytext:備註文字的座標(默認為xy的位置)
- arrowprops:在xy和xytext之間繪製一個箭頭
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlinex=np.arange(-10,11,1)y=x*xplt.title('這是一個示例標題')plt.plot(x,y)# 添加註釋plt.annotate('這是一個示例註釋',xy=(0,1),xytext=(-2,22),arrowprops={'headwidth':10,'facecolor':'r'})plt.show()/<code>
輸出:
4. 設置座標軸名稱-xlabel/ylabel
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlinex=np.arange(1,20)plt.xlabel('示例x軸')plt.ylabel('示例y軸')plt.plot(x,x*x)plt.show()/<code>
輸出:
5. 添加圖例-legend
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlineplt.plot(x,x)plt.plot(x,x*2)plt.plot(x,x*3)plt.plot(x,x*4)# 直接傳入legendplt.legend(['生活','顏值','工作','金錢'])plt.show()/<code>
輸出:
6. 調整顏色-color
傳顏色參數,支持以下幾種方式
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinex=np.arange(1,5)#顏色的幾種方式plt.plot(x,color='g')plt.plot(x+1,color='0.5')plt.plot(x+2,color='#FF00FF')plt.plot(x+3,color=(0.1,0.2,0.3))plt.show()/<code>
輸出:
7. 切換線條樣式-marker
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinex=np.arange(1,5)plt.plot(x,marker='o')plt.plot(x+1,marker='>')plt.plot(x+2,marker='s')plt.show()/<code>
輸出:
8. 顯示數學公式-mathtext
格式如下: \\omega $,中間的將解析出公式中的符號
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlineplt.title('chenqionghe')plt.xlim([1,8])plt.ylim([1,5])plt.text(2,4,r'$ \\alpha \\beta \\pi \\lambda \\omega $',size=25)plt.text(4,4,r'$ \\sin(0)=\\cos(\\frac{\\pi}{2}) $',size=25)plt.text(2,2,r'$ \\lim_{x \\rightarrow y} \\frac{1}{x^3} $',size=25)plt.text(4,2,r'$ \\sqrt[4]{x}=\\sqrt{y} $',size=25)plt.show()/<code>
輸出:
9. 顯示網格-grid
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinex='a','b','c','d'y=[15,30,45,10]plt.grid()# 也可以設置顏色、線條寬度、線條樣式# plt.grid(color='g',linewidth='1',linestyle='-.')plt.plot(x,y)plt.show()/<code>
輸出:
10. 調整座標軸刻度-locator_params
- 同時調整x軸和y軸:plt.locator_params(nbins=20);
- 只調整x軸:plt.locator_params(‘'x',nbins=20);
- 只調整y軸:plt.locator_params(‘'y',nbins=20)。
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinex=np.arange(0,30,1)plt.plot(x,x)# x軸和y軸分別顯示20個plt.locator_params(nbins=20)plt.show()/<code>
輸出:
11. 調整座標軸範圍-axis/xlim/ylim
- axis:[0,5,0,10],x從0到5,y從0到10
- xlim:對應參數有xmin和xmax,分別能調整最大值最小值
- ylim:同xlim用法
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinex=np.arange(0,30,1)plt.plot(x,x*x)#顯示座標軸,plt.axis(),4個數字分別代表x軸和y軸的最小座標,最大座標#調整x為10到25plt.xlim(xmin=10,xmax=25)plt.plot(x,x*x)plt.show()/<code>
輸出:
12. 調整日期自適應-autofmt_xdate
有時候顯示日期會重疊在一起,非常不友好,調用plt.gcf().autofmt_xdate(),將自動調整角度
<code>import numpy as npimport pandas as pdimport matplotlib.pyplot as plt%matplotlib inlinex=pd.date_range('2020/01/01',periods=30)y=np.arange(0,30,1)plt.plot(x,y)plt.gcf().autofmt_xdate()plt.show()/<code>
輸出:
13. 添加雙座標軸-twinx
<code>import numpy as npimport matplotlib.pyplot as plt%matplotlib inlinex=np.arange(1,20)y1=x*xy2=np.log(x)plt.plot(x,y1)# 添加一個座標軸,默認0到1plt.twinx()plt.plot(x,y2,'r')plt.show()/<code>
輸出:
14. 填充區域-fill/fill_beween
fill填充函數區域
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlinex=np.linspace(0,5*np.pi,1000)y1=np.sin(x)y2=np.sin(2*x)plt.plot(x,y1)plt.plot(x,y2)# 填充plt.fill(x,y1,'g')plt.fill(x,y2,'r')plt.title('這是一個示例標題')plt.show()/<code>
輸出:
fill_beween填充函數交叉區域
<code>import numpy as npimport matplotlib.pyplot as plt# 顯示中文plt.rcParams['font.sans-serif'] = [u'SimHei']plt.rcParams['axes.unicode_minus'] = False%matplotlib inlineplt.title('這是一個示例標題')x=np.linspace(0,5*np.pi,1000)y1=np.sin(x)y2=np.sin(2*x)plt.plot(x,y1)plt.plot(x,y2)# 填充plt.fill_between(x,y1,y2,where=y1>y2,interpolate=True)plt.show()/<code>
輸出:
15. 畫一個填充好的形狀-matplotlib.patche
<code>import numpy as npimport matplotlib.pyplot as pltimport matplotlib.patches as mptaches%matplotlib inlinexy1=np.array([0.2,0.2])xy2=np.array([0.2,0.8])xy3=np.array([0.8,0.2])xy4=np.array([0.8,0.8])fig,ax=plt.subplots()#圓形,指定座標和半徑circle=mptaches.Circle(xy1,0.15)ax.add_patch(circle)#長方形rect=mptaches.Rectangle(xy2,0.2,0.1,color='r')ax.add_patch(rect)#多邊形polygon=mptaches.RegularPolygon(xy3,6,0.1,color='g')ax.add_patch(polygon)# 橢圓ellipse=mptaches.Ellipse(xy4,0.4,0.2,color='c')ax.add_patch(ellipse)ax.axis('equal')plt.show()/<code>
輸出:
16. 切換樣式-plt.style.use
matplotlib支持多種樣式,可以通過plt.style.use切換樣式,例如:plt.style.use('ggplot')輸入 plt.style.available 可以查看所有的樣式:
<code>import matplotlib.pyplot as pltplt.style.available/<code>
輸出:
示例代碼,ggplot樣式:
<code>import numpy as npimport matplotlib.pyplot as pltimport matplotlib.patches as mptaches%matplotlib inlineplt.style.use('ggplot')# 新建4個子圖fig,axes=plt.subplots(2,2)ax1,ax2,ax3,ax4=axes.ravel()# 第一個圖x,y=np.random.normal(size=(2,100))ax1.plot(x,y,'o')# 第二個圖x=np.arange(0,10)y=np.arange(0,10)colors=plt.rcParams['axes.prop_cycle']length=np.linspace(0,10,len(colors))for s in length: ax2.plot(x,y+s,'-')# 第三個圖x=np.arange(5)y1,y2,y3=np.random.randint(1,25,size=(3,5))width=0.25ax3.bar(x,y1,width)ax3.bar(x+width,y2,width)ax3.bar(x+2*width,y3,width)# 第四個圖for i,color in enumerate(colors): xy=np.random.normal(size=2)ax4.add_patch(plt.Circle(xy,radius=0.3,color=color['color']))ax4.axis('equal')plt.show()/<code>
輸出:
原文地址:https://www.cnblogs.com/chenqionghe/
校對:林亦霖
—完—
關注清華-青島數據科學研究院官方微信公眾平臺“ AI數據派 ”及姊妹號“ 數據派THU ”獲取更多講座福利及優質內容。
閱讀更多 數據派THU 的文章