本文介绍了带有matplotlib散布的条件颜色的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我有以下Pandas Dataframe，其中列a表示一个伪变量：
我想要做的是在b列的值之后为我的标记添加cmap='jet'颜色，但当a列的值等于1时除外-在本例中，我希望它是灰色。
你知道我该怎么做吗？
推荐答案
您必须将等于1的值标记为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表示 pandas 系列。
这篇关于带有matplotlib散布的条件颜色的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！