使用 seaborn 和 matplotlib 的堆积条形图

Question

我是 pandas 和 matplotlib 的新手，并试图完成以下工作。我有一个如下所示的数据框，它实际上是根据比赛日期列出球员的表现

name          runs  match_date  player_id
Dockrell, G H   0   2018-06-17  3752
Stirling, P R   81  2018-06-17  3586
O'Brien, K J    28  2018-06-17  3391
McCarthy, B J   0   2018-06-17  4563
Poynter, S W    0   2018-06-17  4326
Poynter, S W    2   2018-06-17  4326
McCarthy, B J   0   2018-06-17  4563
Shannon, J N K  5   2018-06-17  4219
Shannon, J N K  6   2018-06-17  4219
Stirling, P R   51  2018-06-17  3586

这是我根据以下代码创建的数据子集

match_performance = dataFrame[['name','runs','match_date','player_id']].sort_values('match_date',ascending=False).groupby('player_id').head(5)
sns.set_context({"figure.figsize": (10, 4)})
ax = sns.barplot(x="name", y="runs",  data=match_performance)
ax.set_xticklabels(ax.get_xticklabels(), rotation=90) 

我需要将其绘制为堆积条形图或分组条形图，以根据数据框中的玩家 ID 显示最近 5 场比赛中玩家的表现，但我不知道如何根据需要绘制这些数据。

Answer 1

虽然这是一个较老的问题，但我在寻找解决方案时发现了它，所以我希望这可以帮助某人。使用 seaborn 实现堆叠条有点棘手，但这应该可以解决问题

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
import matplotlib.patches as mpatches

def stacked_chart_sns(df, x, y, group_by, palette):
    array_of_dfs = []
    w_0 = None
    for u in df[group_by].unique():
        w = df[df[group_by] == u].copy()
        if w_0 is not None:
            w = w.merge(w_0, how='outer').fillna(0)
            w[group_by] = u
            w[y] = w.apply(lambda x: x[y] + x['y_prev'], axis=1)
            w = w.drop(columns=['y_prev'])
        array_of_dfs += [w]
        w_0 = w.drop(columns=[group_by]).rename(columns={y:'y_prev'}).copy()

    patches = []
    for i, d in enumerate(array_of_dfs[::-1]):
        sns.barplot(x=x, y=y, data=d, color=palette[i])
        patches += [mpatches.Patch(label=list(df[group_by].unique())[::-1][i], color=palette[i])]

    plt.legend(handles=patches, loc = 'upper left', ncol=1, labelspacing=1)
    plt.show()

### use it with - for the example data in the question:
stacked_chart_sns(match_performance, 'match_date', 'runs', 'player_id', sns.color_palette("Spectral"))

Answer 2

Seaborn 的创建者 Michael Waskom 在 Twitter 上发布了此消息：

@randyzwitch 我不太喜欢堆积条形图，我建议也许 使用带有 kind=point 的 pointplot / factorplot

— Michael Waskom (@michaelwaskom) 2014 年 9 月 4 日

很遗憾，答案是否定的。 Seaborn 中没有用于绘制堆积条形图的内置函数。