用不同的颜色绘制两个列表

Question

我用类似于stem()的Matlba绘制一个列表（s_n_hat），如下所示：

markerline, stemlines, _ = plt.stem(s_n_hat, '-.')
plt.setp(markerline, 'markerfacecolor', 'b')
plt.setp(baseline, 'color','r', 'linewidth', 2)
plt.show()

在我的实际应用程序中，我想用蓝色绘制命中，用红色绘制未命中，我该怎么做？所以，有些元素应该是蓝色的，有些是红色的。在

假设向量的第一部分是命中，第二部分是未命中，我尝试这样做：

^{pr2}$

我希望第一个元素是蓝色的，其他元素都是红色的，但它们似乎是混合的：

有什么想法吗？在

Answer 1

从documentation：

If no * x * values are provided, the default is (0, 1, ..., len (y) -1)

然后你需要传递一个x，但是这些点是重叠的。比如：

s_n_hat = [1, -1, 1, 1, -1, 1, 1, 1, 1]
x1 = list(range(0, 5))
x2 = list(range(5, 8))
markerline1, stemlines, _ = plt.stem(x1, s_n_hat[0:5], '-.')
plt.setp(markerline1, 'markerfacecolor', 'b')
markerline2, stemlines, _ = plt.stem(x2, s_n_hat[6:9], '-.')
plt.setp(markerline2, 'markerfacecolor', 'r')
plt.show()