我需要使用 matplotlib 或 seaborn 绘制分组数据

Question

我正在尝试绘制国家/地区部门及其销售情况。所以我在不同的列中有国家、部门和他们的销售数字。我需要创建条形图，所以它应该在 x 轴和销售 y 轴上看起来像 country1 及其部门，然后是 country2 等等。

我尝试了 seaborn 的 catplot，但它给了我每个国家/地区的差异图，使用 plotly 的栏它只是将所有部门绘制在一起。我需要根据国家/地区对其进行分组。

感谢任何输入。谢谢

Answer 1

您可以使用'Country' 作为x 和'Department' 作为hue 创建一个barplot()：

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
import seaborn as sns

df = pd.DataFrame({'Country': np.repeat(['Albania', 'Botswana', 'Chili'], 4),
                   'Department': np.tile(['Department1', 'Department2', 'Department3', 'Department4'], 3),
                   'Sales': np.random.randint(10000, 100000, 12)})
sns.barplot(data=df, x='Country', y='Sales', hue='Department')
plt.show()

处理空条的一种方法是创建堆叠条：

df.pivot(index='Country', columns='Department').plot(kind='bar', stacked=True, rot=0)