用不同颜色绘制点

Question

我有一个使用创建的数据框

results = df2[(df2['R'] > 100)].sort(columns='ZR', ascending=False)

我想做

plt.plot(results['ZR'], marker='o')

除了我希望结果['I'] == foo 的点为红色，而结果['I'] != foo 的点为蓝色。

我试过了

firstset = results.ZR[results.I.str.contains('foo')]
secondset = results.ZR[~results.I.str.contains('foo)]
plt.plot(firstset, marker='o', color='red')
plt.plot(secondset, marker='o', color='blue')

但这会从 x 轴 0 开始绘制两半，这不是我需要的。

我会喜欢原来的图表，但有些点是红色的，有些是蓝色的。那不是新的点，也没有改变位置的点。这是原始图表。

Answer 1

首先，在没有任何 X 参数的情况下，plt.show 将其假定为从 1 开始的连续整数。为了使用逻辑运算找到正确的索引，请执行以下操作：

firstIndex = results.index[results.I.str.contains('foo')]
secondIndex = results.index[~results.I.str.contains('foo)]

如果您的原始索引很复杂，请为新索引创建一个虚拟 pandas DataFrame。

newDf = pd.DataFrame(range(len(results)))
firstIndex = newDf.index[results.I.str.contains('foo')]
secondIndex = newDf.index[~results.I.str.contains('foo')]

这保证会创建两个作为1:230 子集的索引。

接下来，将字符串限定符传递给plt.plot 以指定颜色和标记类型。

plt.plot(results['ZR'])
plt.plot(firstIndex, firstset, "bo")
plt.plot(secondIndex, secondset, "ro")

这将使用线条绘制整个带下划线的数据，没有点标记。然后它将用各自的颜色覆盖这两个设置点。