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

Что я могу построить с:

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 изображения, как следующие:

Я думаю, что я мог бы сделать то же самое с морской сеткой, но на самом деле не нашел пути. Я старался:

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

но он дает мне только один участок (без разделения на клиентов) + он не дает мне 3 бара от продавцов:(

Как бы я использовал фазовую сетку, чтобы получить тот же результат, что и мой цикл?

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()

Вы можете добиться чего-то подобного с помощью FacetGrid от Seaborn через

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"])