为每个 plt.step 行赋不同的颜色。可以使用 plt.step 的 color 参数来指定每个线的颜色。可以通过将 color 参数设置为字符串或元组来为每个线分配颜色。下面是一个示例代码:
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(0, 10)
y1 = np.sin(x)
y2 = np.cos(x)
y3 = np.tan(x)
plt.step(x, y1, color='r', label='sin')
plt.step(x, y2, color='g', label='cos')
plt.step(x, y3, color='b', label='tan')
plt.legend()
plt.show()
在这个例子中,我们为每个 plt.step 调用指定了不同的颜色,即红色,绿色和蓝色。这将为每个线创建一个唯一的颜色标识符,并创建一个图例来显示标识符。