python - 为具有特定颜色的 matplotlib 散点图创建颜色编码键
问题描述
这是数据:
import pandas as pd
data = {'letter': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X'], 'color': ['#FF0000', '#FF7F00', '#FFD400', '#FFFF00', '#BFFF00', '#6AFF00', '#00EAFF', '#0095FF', '#0040FF', '#AA00FF', '#FF00AA', '#EDB9B9', '#E7E9B9', '#B9EDE0', '#B9D7ED', '#DCB9ED', '#8F2323', '#8F6A23', '#4F8F23', '#23628F', '#6B238F', '#000000', '#737373', '#CCCCCC'], 'percent': [0.59, 0.569, 0.343, 0.791, 0.099, 0.047, 0.387, 0.232, 0.262, 0.177, 0.522, 0.317, 0.252, 0.617, 0.644, 0.571, 0.382, 0.12, 0.281, 0.855, 0.283, 1.0, 0.844, 0.499], 'score': [0.541, 0.399, 0.625, 0.584, 0.83, 0.859, 0.62, 0.618, 0.545, 0.536, 0.513, 0.563, 0.592, 0.276, 0.037, 0.0, 0.5, 0.653, 0.485, 0.213, 0.44, 0.0, 0.308, 0.35]}
df = pd.DataFrame(data)
# display(df.head())
letter color percent score
0 A #FF0000 0.590 0.541
1 B #FF7F00 0.569 0.399
2 C #FFD400 0.343 0.625
3 D #FFFF00 0.791 0.584
4 E #BFFF00 0.099 0.830
其中最左边的列是索引。
此代码创建一个散点图:
df.plot.scatter(x='percent', y='score', color=df['color'])
在右边,我想要一个键来指定哪个颜色代表哪个字母。理想情况下,它应该是纯色矩形和字母的列表。我还没有找到一个可以使用他们选择的颜色的解决方案,但我需要这种行为,因为将有多个图需要以相同的方式进行颜色编码。
解决方案
您可以使用.legend
Axes 对象的方法:
import matplotlib.lines as mlines
ax = df.plot.scatter(x='percent', y='score', color=df['color'])
ax.legend(handles=[mlines.Line2D([], [], color=r['color'], marker='.', linestyle='None',
markersize=9, label=r['letter'])
for _, r in df.iterrows()
],
ncol=3,
bbox_to_anchor=(1, 1)
)
注意。在我看来,你不应该超过 10 个彩色项目,这会严重影响可读性
由于每种颜色都有一个元素,因此您可以直接注释这些点:
ax = df.plot.scatter(x='percent', y='score', color=df['color'])
for _, r in df.iterrows():
ax.annotate(r['letter'], (r['percent'], r['score']), ha='left', va='bottom')
推荐阅读
- python - 从nodejs产生的python进程读取输出的问题
- php - 从WordPress中的最后一篇文章中获取第一张图片
- c - 计算C中一个句子中单词的出现次数
- c# - Microsoft Office interop Excel 中是否有任何选项隐藏保存对话框?
- javascript - 如何使用一些可选模块构建多包节点模块
- javascript - 如何在 Jquery 选择器中使用变量
- google-apps-script - 问:如何将 URL 域列表转换为完整 url
- google-cloud-platform - terraform 资源 google_compute_instance_group_named_port 始终重新创建
- highcharts - 如何将不同的背景颜色应用于 highcharts 中不同系列的整个工具提示
- c++ - 如何从c ++中的函数返回多个变量