Как отобразить bokeh vbar для вложенных DataFrameGroupBy, агрегированных по сумме? - PullRequest
Новая
       10

Как отобразить bokeh vbar для вложенных DataFrameGroupBy, агрегированных по сумме?

0 голосов
/ 15 октября 2018

Я пишу простой инструмент для игровой аналитики, но у меня была проблема с отображением вложенных фреймов данных в графике Боке.Когда я выбираю основной аргумент категории, например «Сортировать по типу игры» 

     total_players
type               
DM              184
TDM              52


<class 'pandas.core.frame.DataFrame'>

График отображается правильно - https://imgur.com/kTSsvHl

Но когда я пытаюсь выбрать вложенные категории,например, «Сортировать тип игры по карте» и считать по атрибуту «сумма» 

                   total_players
type map                        
DM   Cinema                  116
     Construction             14
     Lighthouse               10
     Ship                     44
TDM  Cinema                   38
     Lighthouse                8
     Ship                      6
<class 'pandas.core.frame.DataFrame'>

График Bokeh будет пустым и не будет отображаться

Я хочу получить что-то вроде этого: https://imgur.com/dP7XBj5, но с использованием метода sum для агрегирования.Набор данных: Набор данных

Код:

Проблема в первом блоке 'if' 

def gamerounds_graph_create(category, attribute, sort_by):
current_feature_set = category.split("_") # transform from "cat1_cat2" label to cat list['cat1', 'cat2'] need for grouping
df = pd.read_sql(GameRounds.select().sql()[0], db.obj.connection()) # 0 - raw sql #TODO split data by chunks
df.to_csv('game_rounds.csv')
if attribute != 'total' and sort_by == '_count':
    """
    Problem in this 'if' block.
    """
    group = df.groupby(current_feature_set, as_index=True).agg({attribute: "sum"})

    source = ColumnDataSource().from_df(group) #Convert df to bokeh format
    ranges = df[current_feature_set[0]].unique()
    del df #Free RAM, df can be big


    plot = figure(plot_height=400, plot_width=1000, x_range=ranges,
              title="{attribute} by {category}".format(attribute=attribute, category=category))
    plot.vbar(source=source, x=category, top=attribute, width=0.8)

else:
    group = df.groupby(current_feature_set)
    factors = df[current_feature_set[0]].unique()
    print(type(group))
    del df #Free RAM, df can be big
    total_factors = len(factors)
    if total_factors < 3 or total_factors > 12:
        total_factors = 12

    palette = palette_dict['Set3'][total_factors]
    source = ColumnDataSource(group) #Convert df to bokeh format
    plot = figure(plot_height=400, plot_width=1000, x_range=group,
              title="{attribute} by {category}".format(attribute=attribute, category=category))

    factors_map = factor_cmap(category, factors=factors, palette=palette) #broken for nested categories
    if attribute == 'total':
        attribute = 'id' #Categories can be counted by numeric attribute only - id
        sort_by = '_count'
    plot.vbar(source=source, x=category, top=attribute.__add__(sort_by), width=0.8,
              fill_color = factors_map
              ) # inbuilt parameter of ColumnDataSource, already computed



plot.legend.orientation = 'horizontal'
return components(plot)

PS Как добавить цвета в vbar для вложенных категорий?

...