Skip to content

plots

Functions:

  • hinton –

    Draw Hinton diagram for visualizing a 2-D matrix.

hinton 

hinton(
    A: Num[ndarray | sparray, "a b"],
    ax: Axes | None = None,
    marker: str = "s",
    update_from: PathCollection | None = None,
) -> Axes

Draw Hinton diagram for visualizing a 2-D matrix.

This has a significant speed-up over the looping version from the matplotlib cookbook, since we utilize the Axes.scatter function, along with some clever math to determine marker scaling to fill the canvas correctly.

Parameters:

  • A

    (Num[ndarray | sparray, 'a b']) –

    2-D matrix to visualize

  • ax

    (Axes | None, default: None ) –

    Optional pre-defined axis to plot on

  • marker

    (str, default: 's' ) –

    Any valid matplotlib.markers option.

Source code in affinis/plots.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
def hinton(
    A:Num[np.ndarray|sparray, "a b"],
    ax:Axes|None=None,
    marker:str='s',
    update_from:PathCollection|None=None,
)->Axes: 
    """Draw Hinton diagram for visualizing a 2-D matrix.

    This has a significant speed-up over the looping version from the
    [matplotlib cookbook](https://matplotlib.org/stable/gallery/specialty_plots/hinton_demo.html),
    since we utilize the `Axes.scatter` function, along with some clever
    math to determine marker scaling to fill the canvas correctly. 

    Args:
      A: 2-D matrix to visualize
      ax: Optional pre-defined axis to plot on
      marker: Any valid [`matplotlib.markers`](https://matplotlib.org/stable/api/markers_api.html#module-matplotlib.markers) option.

    """
    A = A if issparse(A) else coo_array(A)

    i,j = A.coords[1]+0.5, A.coords[0]+0.5
    ax = ax if ax is not None else plt.gca()

    # if update_from is None:
    ax.patch.set_facecolor('gray')
    ax.xaxis.set_major_locator(plt.NullLocator())
    ax.yaxis.set_major_locator(plt.NullLocator())
    ax.set_aspect('equal')
    ax.set_xlim(-0.5,A.shape[1]+0.5)
    ax.set_ylim(-0.5, A.shape[0]+0.5)
    plt.gcf().canvas.draw()
    ax.autoscale_view()
    ax.invert_yaxis()

    ext = ax.get_window_extent()
    base_size = ((ext.width)*(72./plt.gcf().dpi)/(A.shape[1]+2))**2
    size = np.abs(A.data)
    size = base_size*size/size.max()
    color=np.where(A.data>0, 'white', 'black')

    if update_from is not None:
        update_from.set_offsets(np.stack([i, j]).T)
        update_from.set_sizes(size)
        update_from.set_facecolors(color)
        return update_from
    else:
        scat = ax.scatter(
            i,j,
            s=size,
            c=color,
            marker=marker,
            linewidth=0,
        )
        return scat