在matplotlib中,scatter()函數用於繪製散點圖。
下面繪製一個最簡單的散點圖:
import matplotlib.pyplot as plt
import numpy as np
X= np.random.rand(100)
Y= np.random.randn(100)
plt.scatter(X,Y)
plt.show()
我們可以設置一些常用的關鍵字參數:
plt.scatter(X,Y, c ="red", edgecolor = "blue",linewidth =2.5, s= 200, alpha =0.8,lable ="data1")
plt.legend()
plt.show()
其中:
c或color 表示marker點中心區域面的顏色,可以是一個預設的表示顏色的字符串,也可使用RGB顏色(範圍0到1,長度為3的python序列),還可以是和X同長度的數組。
edgecolor 表示marker點邊緣的顏色,默認和中心區域顏色一致。
linewidth 表示marker點邊緣的線寬。
s 表示marker點的尺寸大小,可以是一個標量,也可以是和X同長度的數組。
alpha 表示 marker點的不透明度,0表示完全透明。
lable表示分組標籤,需配合legend()函數使用。
此外,marker 參數 表示marker點的形狀, 默認為"o"。
當 s 也為數組時,可以額外顯示第三個維度的信息,此時散點圖可稱作氣泡圖。
plt.scatter(X,Y,color ="g",linewidth =2.5, s= 100*(X**2+Y**2), alpha =0.8)
plt.show()
同理,color為數組時,也可以顯示額外維度的信息。它們還可以搭配使用:
bubble =plt.scatter(X,Y,c =np.random.rand(100), cmap='RdYlBu',s= 100*(X**2+Y**2))
plt.colorbar(bubble)
plt.show()
注意,和參數cmap配合使用時,顏色參數只能用c,不能用color。
還有其它的colormap可選:
bubble =plt.scatter(X,Y,c =np.random.rand(100), cmap="hot",s= 100*(X**2+Y**2))
下面這種我喜歡:
bubble =plt.scatter(X,Y,c =np.random.rand(100), cmap="nipy_spectral",s= 100*(X**2+Y**2))
plt.colorbar(bubble)
plt.show()
bubble =plt.scatter(X,Y,c =np.random.rand(100), cmap='PuBu_r',s= 100*(X**2+Y**2), alpha =0.8)
plt.colorbar(bubble)
plt.show()
這種也漂亮:
bubble =plt.scatter(X,Y,c =np.random.rand(100), cmap='jet',s= 100*(X**2+Y**2), alpha =0.8)
plt.colorbar(bubble)
plt.colorbar(bubble)plt.show()
依然可以添加其它的圖形元素,使氣泡圖更完整:
plt.title("氣泡圖示例",fontproperties="SimHei",fontsize = 20)
plt.xlabel("x 軸", fontproperties="SimHei")
plt.ylabel("y 軸", fontproperties="SimHei")
閱讀更多 Python可視化編程 的文章