python - 如何使用 Matplotlib 或 Seaborn 根据不同组指定图例
问题描述
我有一个如下所示的数据集:
df = {'tic': {0: 'A',
1: 'AAPL',
2: 'ABC',
3: 'ABT',
4: 'ADBE',
5: 'ADI',
6: 'ADM',
7: 'ADP',
8: 'ADSK',
9: 'AEE'},
'Class': {0: 'Manufacturing',
1: 'Tech',
2: 'Trade',
3: 'Manufacturing',
4: 'Services',
5: 'Tech',
6: 'Manufacturing',
7: 'Services',
8: 'Services',
9: 'Electricity and Transportation'},
'Color': {0: 'blue',
1: 'teal',
2: 'purple',
3: 'blue',
4: 'red',
5: 'teal',
6: 'blue',
7: 'red',
8: 'red',
9: 'orange'},
'Pooled 1': {0: 0.0643791550056838,
1: 0.05022103288830682,
2: 0.039223739393748916,
3: 0.036366693834970217,
4: 0.05772708899447428,
5: 0.05969899935101172,
6: 0.04568101605219955,
7: 0.04542272002937567,
8: 0.07138013872431757,
9: 0.029987722053015278}}
我想生成一个蝙蝠图,其中的值存储在Pooled 1
. 但我想用存储在Color
. 相同的所有条Class
应具有相同的颜色,并应绘制在一起。我只展示了上面数据集的一部分。
我正在使用的代码如下:
fig, axs = plt.subplots(1,1,figsize = (24, 5))
tmp_df = df.sort_values('Class')
plt.bar(np.arange(len(df)), tmp_df['Pooled 1'], color = tmp_df['Color'])
我想有一个图例,其中的名称Class
和颜色来自Color
. 我知道 seaborn 可以做到这一点,barplot
但它不会遵循所需的颜色。而且我不知道为什么,但barplot
绘制数据集需要很长时间。Matplotlib 虽然超级快。
在这种情况下添加图例的最佳方法是什么?提前致谢!
解决方案
您可以为每个类的第一个条分配一个标签。Matplotlib 将使用这些标签来创建图例:
from matplotlib import pyplot as plt
import pandas as pd
import numpy as np
df = pd.DataFrame({'tic': {0: 'A', 1: 'AAPL', 2: 'ABC', 3: 'ABT', 4: 'ADBE', 5: 'ADI', 6: 'ADM', 7: 'ADP', 8: 'ADSK', 9: 'AEE'}, 'Class': {0: 'Manufacturing', 1: 'Tech', 2: 'Trade', 3: 'Manufacturing', 4: 'Services', 5: 'Tech', 6: 'Manufacturing', 7: 'Services', 8: 'Services', 9: 'Electricity and Transportation'}, 'Color': {0: 'blue', 1: 'teal', 2: 'purple', 3: 'blue', 4: 'red', 5: 'teal', 6: 'blue', 7: 'red', 8: 'red', 9: 'orange'}, 'Pooled 1': {0: 0.0643791550056838, 1: 0.05022103288830682, 2: 0.039223739393748916, 3: 0.036366693834970217, 4: 0.05772708899447428, 5: 0.05969899935101172, 6: 0.04568101605219955, 7: 0.04542272002937567, 8: 0.07138013872431757, 9: 0.029987722053015278}})
fig, ax = plt.subplots(1, 1, figsize=(14, 5))
tmp_df = df.sort_values('Class')
bars = ax.bar(tmp_df['tic'], tmp_df['Pooled 1'], color=tmp_df['Color'])
prev = None
for cl, color, bar in zip(tmp_df['Class'], tmp_df['Color'], bars):
if cl != prev:
bar.set_label(cl)
prev = cl
ax.margins(x=0.01)
ax.legend(title='Class', bbox_to_anchor=(1.01, 1.01), loc='upper left')
plt.tight_layout()
plt.show()
PS:请注意,您也可以使用 Seaborn 并让着色自动进行:
import seaborn as sns
sns.barplot(data=tmp_df, x='tic', y='Pooled 1', hue='Class', palette='tab10', dodge=False, saturation=1, ax=ax)
推荐阅读
- react-native - 如何在 ios stimulator 中查看 react-native 创建的领域数据库
- mysql - MySQL:锁定表与自动提交与开始交易
- sql-server - SQL Server 按具有英寸大小的列排序
- javascript - 如何在 Chrome 扩展中将 popup.html 调整为全屏
- angular - Angular 可观察订阅问题
- javascript - 在 JavaScript 中使用变量访问数组索引
- java - 如何在 Java Pulse 中制作多边形对象(如 Atari 游戏“冒险”中的圣杯)
- python - BeautifulSoup 如何找到 XML 标签?
- r - R:如何跨区域可视化给定列中的前 5 个值
- c# - 通用接口和类/向上转换不起作用