gpt4 book ai didi

python - 使用正确的标签 python 绘制图例

转载 作者:太空宇宙 更新时间:2023-11-03 11:47:38 25 4
gpt4 key购买 nike

我有一个数据框,我想绘制一个带有“A”、“B”和“C”的图例,但是,我只生成了一个带有“A”标签的图例:

data = {'A1_mean': [0.457, 1],
'A2_median': [0.391,1],
'A3_range': [0.645,1],
'A4_std': [0.111,1],
'B1_mean': [0.132,3],
'B2_median': [0.10,3],
'B3_range': [0.244,3],
'B4_std': [0.297,3],
'C1_mean': [0.286,2],
'C2_median': [0.231,2],
'C3_range': [0.554,2],
'C4_std': [0.147,2]}
df = pd.DataFrame(data).T
color = {1:'red',2:'green',3:'blue'}
ax=df[0].plot(kind='bar',color=df[1].map(color).tolist())
ax.legend(['A','B','C'])

给出:

enter image description here

我怎样才能改变这个,以便我有一个带有 A B 和 C 的图例,具有适当的颜色(A:红色,B:蓝色,C:绿色)?

最佳答案

根据 Legend guide您可以将代理艺术家放在图例中:

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.patches as mpatches

data = {'A1_mean': [0.457, 1],
'A2_median': [0.391,1],
'A3_range': [0.645,1],
'A4_std': [0.111,1],
'B1_mean': [0.132,3],
'B2_median': [0.10,3],
'B3_range': [0.244,3],
'B4_std': [0.297,3],
'C1_mean': [0.286,2],
'C2_median': [0.231,2],
'C3_range': [0.554,2],
'C4_std': [0.147,2]}
df = pd.DataFrame(data).T
color = {1:'red',2:'green',3:'blue'}
labels = ['A','C','B']
fig, ax = plt.subplots()
df[0].plot(ax=ax, kind='bar', color=df[1].map(color))

handles = []
for i, c in color.items():
handles.append(mpatches.Patch(color=c, label=labels[i-1]))
plt.legend(handles=handles, loc='best')
# auto-rotate xtick labels
fig.autofmt_xdate()
plt.show()

enter image description here

关于python - 使用正确的标签 python 绘制图例,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35413523/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com