多列图 Seaborn 和 Matplotlib

Question

我有这个数据表：

Date, City, Sales, PSale1, PSale2

其中 PSale1 和 PSale2 是销售的预测价格。

数据集如下所示：

Date,City,Sales,PSale1,PSale2
01/01,NYC,200,300,178
01/01,SF,300,140,100
02/01,NYC,400,410,33
02/01,SF,42,24,60

我想在 seaborn 中绘制这些数据，我正在绘制一个图表，该图表将在 x 轴上显示日期，PSale1 和 PSale2 作为 y 轴上每个城市的条形图，但我很困惑如何使用这样的数据。

据我所知，seaborn 不允许在任何轴上绘制两个变量，我应该如何处理这种情况？

Answer 1

• 将数据帧转换为长格式
• 使用seaborn.FacetGrid 绘图

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

data = {'Date': ['01/01', '01/01', '02/01', '02/01'],
        'City': ['NYC', 'SF', 'NYC', 'SF'],
        'Sales': [200, 300, 400, 42],
        'PSale1': [300, 140, 410, 24],
        'PSale2': [178, 100, 33, 60]}

df = pd.DataFrame(data)

# convert to long
dfl = df.set_index(['Date', 'City', 'Sales']).stack().reset_index().rename(columns={'level_3': 'pred', 0: 'value'})

# plot
g = sns.FacetGrid(dfl, row='City')
g.map(sns.barplot, 'Date', 'value', 'pred').add_legend()
plt.show()

---

• 合二为一

# shape the dataframe
dfc = df.drop(columns=['Sales']).set_index(['Date', 'City']).stack().unstack(level=1)
dfc.columns.name = None

# plot
dfc.plot.bar(stacked=True)
plt.xlabel('Date: Predicted')
plt.show()