Matplotlib 绘制公式变化的图形

Question

所以我基本上有一些数据是这样的：x = [0, 5, 12, 17] 和y = [0, 1, 0, 0]（我的意思是那些真的很长的列表）。现在我想绘制一个图，其值为0，从0 到5，然后从5 到12 -> 1，然后从12 到17 0 的值。现在我可以通过生成这样的数组来做到这一点： x = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17] 和 y = [0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, ...., 0] 和只是 plt.plot(x,y)。但是对于大型数组我猜这效率不高，所以我想问一下是否有matplotlib方法可以打印指定范围内的指定值或函数公式？ （例如 sin(x) 从 0 到 5 以及其他更多内容）

Answer 1

可以使用plt.step 绘制示例数据。在这种情况下，参数where 必须是where="post"。

import matplotlib.pyplot as plt

x = [0, 5, 12, 17]
y = [0, 1, 0, 0]

plt.step(x, y, where='post', label='step')
plt.xticks(x)

plt.show()