熊猫散点图:多种颜色和背景

时间:2015-01-29 23:53:14

标签: python matplotlib pandas

我正在尝试在scores中创建数据的散点图,其中threshold以上的值为红色。我有相应的颜色显示问题。我还想为labels中指定的索引绘制图形的背景,这是一个0和1的数据帧长度的列表。下面是我到目前为止的代码。在此先感谢您的帮助!

import pandas
...

values = pandas.DataFrame({"scores":scores, "labels":labels, "indices":range(len(labels))})

# plot alerts in red, others in black
alertValues = values.iloc[scores[scores >= threshold].index]

ax = alertValues.plot(kind="scatter", x="indices", y="scores", c='r',
  marker='.', s=5, alpha=0.5)
values.plot(kind="scatter", x="indices", y="scores", c='r',
  marker='.', s=5, alpha=0.5, ax=ax)

# show plot
plt.title(fileName)
plt.show()

1 个答案:

答案 0 :(得分:1)

问题在于从pandas DataFrame调用plot函数。将pyplot与列表一起使用会更好:

plt.figure()
plt.scatter(indices, scores)
plt.scatter(fooIndices, fooScores)
plt.show()