使用matplotlib怎么繪制一個階梯圖

這篇文章將為大家詳細講解有關使用matplotlib怎么繪制一個階梯圖，文章內容質量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關知識有一定的了解。

step函數概述

step函數用于繪制階梯圖。

根據源碼可知，step函數是對plot函數的輕量級封裝，很多概念和用法與plot函數非常相似。

def step(self, x, y, *args, where='pre', data=None, **kwargs):
 cbook._check_in_list(('pre', 'post', 'mid'), where=where)
 kwargs['drawstyle'] = 'steps-' + where
 return self.plot(x, y, *args, data=data, **kwargs)

step函數簽名：

matplotlib.pyplot.step(x, y, *args, where='pre', data=None, **kwargs)

step函數調用簽名：

step(x, y, [fmt], *, data=None, where='pre', **kwargs)
step(x, y, [fmt], x2, y2, [fmt2], ..., *, where='pre', **kwargs)

其中：

• x：類數組結構，一維x軸坐標序列。一般假設x軸坐標均勻遞增。必備參數。

• y：類數組結構，一維y軸坐標序列。必備參數。

• fmt：格式字符串，與plot函數的fmt參數類似。可選參數。官方建議只設置顏色格式。

• data：可索引數據，類似于plot函數。可選參數。

• **kwargs：類似于plot函數。

• where ：設置階梯所在位置，取值范圍為{'pre', 'post', 'mid'}，默認值為'pre'。

 案例：使用step函數和plot函數演示不同where參數的效果

通過案例可知，step函數可以認為是plot函數繪制階梯圖的一個特例。

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(14)
y = np.sin(x / 2)

plt.figure(figsize=(12,5))
plt.subplot(121)
plt.step(x, y + 2, label='pre (default)')
plt.plot(x, y + 2, 'o--', color='grey', alpha=0.3)

plt.step(x, y + 1, where='mid', label='mid')
plt.plot(x, y + 1, 'o--', color='grey', alpha=0.3)

plt.step(x, y, where='post', label='post')
plt.plot(x, y, 'o--', color='grey', alpha=0.3)

plt.grid(axis='x', color='0.95')
plt.legend(title='Parameter where:')
plt.title('plt.step(where=...)')

plt.subplot(122)
plt.plot(x, y + 2, drawstyle='steps', label='steps (=steps-pre)')
plt.plot(x, y + 2, 'o--', color='grey', alpha=0.3)

plt.plot(x, y + 1, drawstyle='steps-mid', label='steps-mid')
plt.plot(x, y + 1, 'o--', color='grey', alpha=0.3)

plt.plot(x, y, drawstyle='steps-post', label='steps-post')
plt.plot(x, y, 'o--', color='grey', alpha=0.3)

plt.grid(axis='x', color='0.95')
plt.legend(title='Parameter drawstyle:')
plt.title('plt.plot(drawstyle=...)')
plt.show()

關于使用matplotlib怎么繪制一個階梯圖就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。