繁体   English   中英

如何使用 matplotlib 在 python 条形图中绘制可变数量的集合?

[英]How do I plot a variable number of sets in a python bar chart using matplotlib?

我正在尝试编写一个代码,该代码基于输入数据集,可以打印与数据集对应的分组条形图。 数据集可以有不同的场景,每个场景可以有不同的方法。 我想为每个场景打印一组条形图(代表不同的方法)(示例见下图)。

条形图示例

问题是方法的数量和场景的数量都不是固定的。 我不知道如何让我的代码适应不同的情况。 到目前为止,我的代码读取数据集并创建一个字典,其中键是不同的方法,值是场景的字典作为键,最终值作为值。

例子

my_dict = {method1: {scenario1: value1, scenario2: value2, ...}, method2: {scenario1: value1, scenario2: value2, ...}}

我只是不知道如何使用这本字典来绘制我的数据。 我知道通常我需要定义我的类别列表(上面示例中的方法),但由于列表(方法)的数量是可变的,我被卡住了,不知道如何从这里移动。 任何帮助将不胜感激!

我不知道如何使用我的嵌套字典来定义要绘制的不同集合。

import seaborn as sns

my_dict = pd.DataFrame({'method1': {'scenario1': 10, 'scenario2': 20, 'scenario3': 1}, 'method2': {'scenario1': 15, 'scenario2': 5, 'scenario3': 1.5}})

my_dict = my_dict.reset_index()

df = my_dict.melt(id_vars='index')

g = sns.catplot(data=df, x="index", y="value", hue="variable", kind="bar")

这给了我以下情节:

在此处输入图像描述

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM