def polar_map(hist: Histogram2D, ax: Axes, *, show_zero: bool = True, show_colorbar: bool = True, **kwargs): """Polar map of polar histograms. Similar to map, but supports less parameters.""" data = get_data(hist, cumulative=False, flatten=True, density=kwargs.pop("density", False)) cmap = _get_cmap(kwargs) norm, cmap_data = _get_cmap_data(data, kwargs) colors = cmap(cmap_data) rpos, phipos = (arr.flatten() for arr in hist.get_bin_left_edges()) dr, dphi = (arr.flatten() for arr in hist.get_bin_widths()) rmax, _ = (arr.flatten() for arr in hist.get_bin_right_edges()) bar_args = {} if "zorder" in kwargs: bar_args["zorder"] = kwargs.pop("zorder") alphas = _get_alpha_data(cmap_data, kwargs) if np.isscalar(alphas): alphas = np.ones_like(data) * alphas for i in range(len(rpos)): if data[i] > 0 or show_zero: bin_color = colors[i] # TODO: align = "edge" bars = ax.bar(phipos[i], dr[i], width=dphi[i], bottom=rpos[i], align='edge', color=bin_color, edgecolor=kwargs.get("grid_color", cmap(0.5)), lw=kwargs.get("lw", 0.5), alpha=alphas[i], **bar_args) ax.set_rmax(rmax.max()) if show_colorbar: _add_colorbar(ax, cmap, cmap_data, norm)
def map(h2: Histogram2D, *, show_zero: bool = True, show_values: bool = False, **kwargs) -> dict: """Heat-map of two-dimensional histogram.""" vega = _create_figure(kwargs) values_arr = get_data(h2, kwargs.pop("density", None), kwargs.pop("cumulative", None)) values = values_arr.tolist() value_format = get_value_format(kwargs.pop("value_format", None)) _add_title(h2, vega, kwargs) _create_scales(h2, vega, kwargs) _create_axes(h2, vega, kwargs) _create_cmap_scale(values_arr, vega, kwargs) _create_colorbar(vega, kwargs) x = h2.get_bin_centers(0) y = h2.get_bin_centers(1) x1 = h2.get_bin_left_edges(0) x2 = h2.get_bin_right_edges(0) y1 = h2.get_bin_left_edges(1) y2 = h2.get_bin_right_edges(1) data = [] for i in range(h2.shape[0]): for j in range(h2.shape[1]): if not show_zero and values[i][j] == 0: continue item = { "x": float(x[i]), "x1": float(x1[i]), "x2": float(x2[i]), "y": float(y[j]), "y1": float(y1[j]), "y2": float(y2[j]), "c": float(values[i][j]) } if show_values: item["label"] = value_format(values[i][j]) data.append(item) vega["data"] = [{ "name": "table", "values": data }] vega["marks"] = [ { "type": "rect", "from": {"data": "table"}, "encode": { "enter": { "x": {"scale": "xscale", "field": "x1"}, "x2": {"scale": "xscale", "field": "x2"}, "y": {"scale": "yscale", "field": "y1"}, "y2": {"scale": "yscale", "field": "y2"}, "fill": {"scale": "color", "field": "c"}, "stroke": {"value": 0}, # "strokeWidth": {"value": 0}, # "fillColor": {"value": "#ffff00"} }, "update": { "fillOpacity": {"value": kwargs.pop("alpha", 1)} }, # "hover": { # "fillOpacity": {"value": 0.5} # } } } ] if show_values: vega["marks"].append( { "type": "text", "from": {"data": "table"}, "encode": { "enter": { "align": {"value": "center"}, "baseline": {"value": "middle"}, "fontSize": {"value": 13}, "fontWeight": {"value": "bold"}, "text": {"field": "label"}, "x": {"scale": "xscale", "field": "x"}, "y": {"scale": "yscale", "field": "y"}, } } } ) return vega
def map(h2: Histogram2D, *, show_zero: bool = True, show_values: bool = False, **kwargs) -> dict: """Heat-map of two-dimensional histogram.""" vega = _create_figure(kwargs) values_arr = get_data(h2, kwargs.pop("density", None), kwargs.pop("cumulative", None)) values = values_arr.tolist() value_format = get_value_format(kwargs.pop("value_format", None)) _add_title(h2, vega, kwargs) _create_scales(h2, vega, kwargs) _create_axes(h2, vega, kwargs) _create_cmap_scale(values_arr, vega, kwargs) _create_colorbar(vega, kwargs) x = h2.get_bin_centers(0) y = h2.get_bin_centers(1) x1 = h2.get_bin_left_edges(0) x2 = h2.get_bin_right_edges(0) y1 = h2.get_bin_left_edges(1) y2 = h2.get_bin_right_edges(1) data = [] for i in range(h2.shape[0]): for j in range(h2.shape[1]): if not show_zero and values[i][j] == 0: continue item = { "x": float(x[i]), "x1": float(x1[i]), "x2": float(x2[i]), "y": float(y[j]), "y1": float(y1[j]), "y2": float(y2[j]), "c": float(values[i][j]) } if show_values: item["label"] = value_format(values[i][j]) data.append(item) vega["data"] = [{ "name": "table", "values": data }] vega["marks"] = [ { "type": "rect", "from": {"data": "table"}, "encode": { "enter": { "x": {"scale": "xscale", "field": "x1"}, "x2": {"scale": "xscale", "field": "x2"}, "y": {"scale": "yscale", "field": "y1"}, "y2": {"scale": "yscale", "field": "y2"}, "fill": {"scale": "color", "field": "c"}, "stroke": {"value": 0}, # "strokeWidth": {"value": 0}, # "fillColor": {"value": "#ffff00"} }, "update": { "fillOpacity": {"value": kwargs.pop("alpha", 1)} }, # "hover": { # "fillOpacity": {"value": 0.5} # } } } ] if show_values: vega["marks"].append( { "type": "text", "from": {"data": "table"}, "encode": { "enter": { "align": {"value": "center"}, "baseline": {"value": "middle"}, "fontSize": {"value": 13}, "fontWeight": {"value": "bold"}, "text": {"field": "label"}, "x": {"scale": "xscale", "field": "x"}, "y": {"scale": "yscale", "field": "y"}, } } } ) return vega