Python Seaborn Facetgrid

Question

我对每个客户的业务员都有绿色/橙色/红色的计数： 像

    sellor  customer    red green   orange
0   73  c1  5   96  15
1   77  c1  88  18  79
2   97  c1  58  59  71

我可以用：p构建

df = pd.DataFrame()
df["sellor"]   = np.random.randint(0,100, 20)
df["customer"]   = ["c1"]*5 + ["C2"]*5 + ["C3"]*5 + ["c4"]*5
df["red"] = np.random.randint(0,100, 20)
df["green"] = np.random.randint(0,100, 20)
df["orange"] = np.random.randint(0,100, 20)

df.sellor = df.sellor.astype("category")
df.customer = df.customer.astype("category")

现在我想将数据表示为图形： 现在我开始：

for customer in df.customer.unique():
    df[df.customer==customer].plot.bar(title=customer)

哪一张给我4张图像，如下图所示：

我想我可以对Seaborn facetgrid进行同样的操作，但并没有真正找到方法。 我尝试过：

sns.barplot(data=df, x="sellor", y="red", hue="customer")

但是它只给我一个地块（不按客户细分）+不能给我我卖方的三个小条:(

我将如何使用facetgrid获得与循环相同的结果？

Answer 1

在不使用seaborn的情况下，您可以将条形图绘制到同一图形的不同子图上，

import matplotlib.pyplot as plt

u = df.customer.unique()
fig, axes = plt.subplots(ncols=len(u), figsize=(10,3))

for customer, ax in zip(u, axes):
    df[df.customer==customer].plot.bar(x="sellor", title=customer, ax =ax )

plt.tight_layout()    
plt.show()

您可以通过seaborn的FacetGrid实现类似的功能

import seaborn as sns
g = sns.FacetGrid(data=df, col="customer", col_order=["c1", "C2", "C3", "c4"])

def plot(*args,**kwargs):
    kwargs["data"].plot(kind="bar", x="sellor", ax=plt.gca())
g.map_dataframe(plot)
plt.show()

最后，要更改此处使用的颜色以匹配数据框名称的颜色，您可以设置颜色周期，例如

plt.rcParams["axes.prop_cycle"] = plt.cycler("color", ["red", "green", "orange"])