使用matplotlib散布的条件颜色
使用matplotlib散布的条件颜色
提问于 2017-02-25 16:16:06
我有以下Pandas Dataframe，其中列a表示一个伪变量：
我想要做的是在b列的值之后给我的标记一个cmap='jet'颜色，除非a列中的值等于1-在本例中我希望它是灰色。
你知道我该怎么做吗？
Stack Overflow用户
回答已采纳
发布于 2017-02-25 16:59:29
您必须将您的值标记为等于1，并绘制：
import matplotlib.pyplot as plt
import numpy as np
# test data
t = np.linspace(0, 2 * np.pi, 30)
x = np.sin(t)
x[3] = 1
y = np.cos(t)
# indices for 'bad' values
indices = x == 1
# calc colors from jet cmap
cmap = plt.get_cmap('jet')
colors = cmap((y - y.min()) / y.ptp())
# normal values
plt.scatter(t[~indices], x[~indices], c = colors[~indices], cmap = cmap)
# bad values
plt.scatter(t[indices], x[indices], c = 'grey')
plt.show()
数组t，x，y表示熊猫序列。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/42453649
复制相关文章
相似问题
腾讯云开发者
