def heatmap_builder(data_id, **inputs): """ Builder function for :plotly:`plotly.graph_objects.Heatmap <plotly.graph_objects.Heatmap>` :param data_id: integer string identifier for a D-Tale process's data :type data_id: str :param inputs: Optional keyword arguments containing the following information: - x: column to be used as x-axis of chart - y: column to be used as y-axis of chart - z: column to use for the Z-Axis - agg: points to a specific function that can be applied to :func: pandas.core.groupby.DataFrameGroupBy :type inputs: dict :return: heatmap :rtype: :plotly:`plotly.graph_objects.Heatmap <plotly.graph_objects.Heatmap>` """ try: if not valid_chart(**inputs): return None raw_data = global_state.get_data(data_id) wrapper = chart_wrapper(data_id, raw_data, inputs) hm_kwargs = dict(hoverongaps=False, colorscale='Greens', showscale=True, hoverinfo='x+y+z') x, y, z, agg = (inputs.get(p) for p in ['x', 'y', 'z', 'agg']) y = y[0] data = retrieve_chart_data(raw_data, x, y, z) x_title = update_label_for_freq(x) y_title = update_label_for_freq(y) z_title = z data = data.sort_values([x, y]) check_all_nan(data) dupe_cols = [x, y] if agg is not None: z_title = '{} ({})'.format(z_title, AGGS[agg]) if agg == 'corr': data = data.dropna() data = data.set_index([x, y]).unstack().corr() data = data.stack().reset_index(0, drop=True) y_title = x_title dupe_cols = [ '{}{}'.format(col, i) for i, col in enumerate(data.index.names) ] [x, y] = dupe_cols data.index.names = dupe_cols data = data.reset_index() data.loc[data[x] == data[y], z] = np.nan hm_kwargs = dict_merge( hm_kwargs, dict(colorscale=[[0, 'red'], [0.5, 'yellow'], [1.0, 'green']], zmin=-1, zmax=1)) else: data = build_agg_data(data, x, y, inputs, agg, z=z) if not len(data): raise Exception('No data returned for this computation!') check_exceptions( data[dupe_cols], agg != 'corr', data_limit=40000, limit_msg= 'Heatmap exceeds {} cells, cannot render. Please apply filter...') dtypes = { c: classify_type(dtype) for c, dtype in get_dtypes(data).items() } data_f, _ = chart_formatters(data) data = data_f.format_df(data) data = data.sort_values([x, y]) data = data.set_index([x, y]) data = data.unstack(0)[z] x_data = weekday_tick_handler(data.columns, x) y_data = weekday_tick_handler(data.index.values, y) heat_data = data.values x_axis = dict_merge({ 'title': x_title, 'tickangle': -20 }, build_spaced_ticks(x_data)) if dtypes.get(x) == 'I': x_axis['tickformat'] = '.0f' y_axis = dict_merge({ 'title': y_title, 'tickangle': -20 }, build_spaced_ticks(y_data)) if dtypes.get(y) == 'I': y_axis['tickformat'] = '.0f' hovertemplate = ''.join([ x_title, ': %{customdata[0]}<br>', y_title, ': %{customdata[1]}<br>', z_title, ': %{z}<extra></extra>' ]) hm_kwargs = dict_merge( hm_kwargs, dict(z=heat_data, colorbar={'title': z_title}, hoverinfo='x+y+z', hovertemplate=hovertemplate, customdata=[[[xd, yd] for xd in x_data] for yd in y_data])) return wrapper( dcc.Graph(id='heatmap-graph-{}'.format(y), style={ 'margin-right': 'auto', 'margin-left': 'auto', 'height': 600 }, figure=dict(data=[go.Heatmap(**hm_kwargs)], layout=build_layout( dict_merge( dict(xaxis=x_axis, yaxis=y_axis, xaxis_zeroline=False, yaxis_zeroline=False), build_title(x, y, z=z, agg=agg)))))) except BaseException as e: return build_error(str(e), str(traceback.format_exc()))
def test_weekday_tick_handler(unittest): unittest.assertEqual( chart_utils.weekday_tick_handler([1, 2, 3], "date|WD"), ["Tues", "Wed", "Thur"]) unittest.assertEqual(chart_utils.weekday_tick_handler([1, 2, 3], "foo"), [1, 2, 3])
def test_weekday_tick_handler(unittest): unittest.assertEqual( chart_utils.weekday_tick_handler([1, 2, 3], 'date|WD'), ['Tues', 'Wed', 'Thur']) unittest.assertEqual(chart_utils.weekday_tick_handler([1, 2, 3], 'foo'), [1, 2, 3])