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