Python語言具有簡潔、高效、易學等優點,近年在數據分析和圖形繪制等領域廣泛應用。階梯圖是常見的數據可視化圖形之一,Python也有相應的庫來實現階梯圖的繪制。
import matplotlib.pyplot as plt import numpy as np x = np.array([0, 1, 2, 3]) y = np.array([0, 2, 4, 3]) plt.step(x, y, where='post') plt.show()
上述代碼實現了一個簡單的階梯圖。np.array()函數用于將列表轉換成數組,plt.step()函數則用于畫階梯圖,其中where='post'指定了從左往右畫,從0點開始。執行后,可以得到如下圖:
4 | * | * |---* * | -- 0----1----2----3
在階梯圖中,每一步可以看作是一段線段,且只有兩種水平:豎直和水平。通過改變where參數,我們可以實現不同的階梯圖。where參數只有兩種選擇取值:
- "pre":默認值,表示從上一步開始,到當前步結束,沿垂直線繪制整個階梯;
- "post":表示從當前步開始,到下一步結束,沿垂直線繪制整個階梯。
如果需要同時繪制多條階梯線,可以將數據放在二維數組中,并指定labels參數。代碼示例如下:
import matplotlib.pyplot as plt import numpy as np data = np.array([[0, 1, 2, 3], [0, 2, 4, 3], [1, 3, 2, 5]]) labels = ['line1', 'line2', 'line3'] for i in range(len(data)): plt.step(data[i], label=labels[i]) plt.legend() plt.show()
上述代碼可以繪制多條階梯線,并標注不同線條的標簽。執行后,可以得到如下圖:
5 | * *--- | | | |-|* | | | | | 0-1-2--3
總之,Python中的階梯圖繪制不僅實用簡單,而且功能強大。有了這個工具,數據可視化變得輕而易舉。
上一篇css中下邊框顏色
下一篇php iis 一鍵安裝