Python繪製簡單的折線圖與散點圖

繪製簡單的折線圖

用Python繪製簡單的折線圖,需要用到一個數學繪圖庫matplotlib
。安裝這個庫很簡單,直接pip install matplotlib。

下面我們繪製一個簡單的折線圖,是數字1,2,3,4與他們的平方的對應關係。
首先導入模塊matplotlib.pyplot,模塊pyplog包含很多用於生成圖表的函數。

import matplotlib.pyplot as plt

然後創建一個列表存儲1,2,3,4,5對應的平方值

squares=[1,4,9,16,25]

然後將這個列表傳給函數plot(),這個函數嘗試根據這些數字繪製出有意義的圖形,然後再使用plt.show()函數打開matplotlib查看器,並顯示繪製的圖形

plt.plot(squares)
plt.show()

這時候我們運行一下,就可以得到一個這樣的圖形初步繪製的圖形

校正圖形

現在,我們確實繪製出了一個圖形,但是,仔細觀察一下就會發現,這個圖形是不正確的,橫座標4對應的平方怎麼能是25呢,這是爲什麼呢。
當我們像plot()提供一系列數字時,他假設傳給他的第一個數據對應的x座標值爲0,但是,我們傳給他的第一個數字是1,並不是0。這種情況下,我們可以通過修改我們傳入的值,也就是列表squares的值,讓它的第一個數據爲0來糾正。

squares=[0,1,4,9,16,25]

或者,爲了改變這種默認行爲,我們可以給plot()同屬提供輸入值和輸出值。
我們增加一個列表,存儲[1,2,3,4,5],這樣就不用修改squares了

x_values=[1,2,3,4,5]

然後對plot函數進行傳參

plt.plot(x_values,squares)

這個時候,我們得到的圖形如下
校正後的圖像

繪製散點圖

用scatter()函數可以繪製單個點的散點圖,像函數傳遞一個x座標,y座標,它將在指定位置繪製一個點

plt.scatter(2,4)

單點
如果要繪製一系列的散點,我們可以給scatter函數傳遞兩個列表

x_values=[1,2,3,4,5]
y_values=[1,4,9,16,25]
plt.plot(x_values,y_values)

結果如下散點圖
當我們需要計算的點比較多時,比如,要繪製1-100的平方,我們可以這樣寫
使用range函數生成一個整數列表作爲x座標,然後使用for循環進行運算得到對應的y座標

x_values=list(range(1,100))
y_values=[i**2 for i in x_values]
plt.scatter(x_values,y_values)

得到的結果如下圖
1-100的平方散點圖
如果你嫌棄這個圖不好看,可以通過一系列設定讓它變得好看些,漸變的顏色看起來比較好看,我們可以讓它指定爲橘紅色,然後從淺到深漸變。並且讓這些點變得小一點。

plt.scatter(x_values,y_values,c=y_values,cmap='Oranges',s=30)

效果如下
調整後

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章