示例

在繪圖時經(jīng)常遇到類似區(qū)域填充的問題，比如對于y = sin ? x y=\sin xy=sinx函數(shù)，想填充其與X軸所圍成的區(qū)間，那么就可以使用fill_between函數(shù)。

import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 2, 0.01)
y = np.sin(2 * np.pi * x)
plt.plot(x, y, color='black')
plt.fill_between(x, y)
plt.grid()
plt.show()

效果如下

參數(shù)

fill_between參數(shù)如下，其中x, y1, y2都是數(shù)組，表示希望覆蓋的范圍，具體邏輯是，對于某點xi?，將對應(yīng)的[y1i?，y2i?]范圍內(nèi)涂上顏色。在上面的示例中，只設(shè)置了x和y1，而y2默認為0，所以繪制的就是y1到0范圍內(nèi)的值。

fill_between(x, y1, y2=0, where=None, interpolate=False, step=None, *,**kwargs)

其他參數(shù)含義如下

• where 可以指定參與繪圖的范圍，一般是一個布爾型數(shù)組
• interpolate 為True時采用插值
• step 可選’pre’, ‘post’, ‘mid’，表示步進方案

plt.plot(x, y, color='black')
plt.fill_between(x, y, where=y > 0, facecolor='green', alpha=.5)
plt.fill_between(x, y, where=y < 0, facecolor='red', alpha=.5)
plt.show()

結(jié)果如圖

回歸圖

有了這種填充工具，就可以做出類似seaborn中的lmplot，這種圖形在數(shù)據(jù)擬合時十分有用，可以在除了擬合線之外，再將數(shù)據(jù)的分布范圍標出。

首先創(chuàng)造一組帶有誤差的y = 2x + 1y=2x+1y=2x+1數(shù)據(jù)，并通過最小二乘法得到其擬合參數(shù)。

x = np.arange(0, 2, 0.02)
err = np.random.rand(100)/2
y = 2*x + 1 + err
# 構(gòu)造并調(diào)用最小二乘法
A = np.array([x, np.ones_like(x)]).T
kb, res, _, _ = np.linalg.lstsq(A, y)

下面就對擬合結(jié)果進行繪制，首先根據(jù)擬合出來的k , bk, bk,b，來得到趨勢線Y = kx + bY=kx+bY=kx+b，然后調(diào)用繪圖函數(shù)，對原始數(shù)據(jù)、趨勢線、分布區(qū)間進行繪制

Y = kb[0]*x + kb[1]
dx = (x-x.mean())**2
E = x.std() * np.sqrt(1/len(x) + dx / np.sum(dx))
plt.scatter(x, y, marker='.')   # 原始數(shù)據(jù)
plt.plot(x, Y)                  # 趨勢線
plt.fill_between(x, Y - E, Y + E, alpha=0.5)
plt.show()

效果如下

到此這篇關(guān)于python matplotlib用面積填充實現(xiàn)lmplot的代碼示例的文章就介紹到這了,更多相關(guān)python matplotlib實現(xiàn)lmplot內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評論