Пример #1
0
def add_circles(treeplot, nodes, colors="g", size=15, xoff=0, yoff=0, vis=True):
    """
    Draw circles on plot

    Args:
        nodes: A node object or list of Node objects or label or list of labels
        colors: Str or list of strs. Colors of the circles. Optional,
          defaults to 'g' (green)
        size (float): Size of the circles. Optional, defaults to 15
        xoff, yoff (float): X and Y offset. Optional, defaults to 0.

    """
    points = xy(treeplot, nodes)
    trans = offset_copy(
        treeplot.transData, fig=treeplot.figure, x=xoff, y=yoff, units='points'
        )

    col = CircleCollection(
        sizes=(pi*size*size*0.25,),
        offsets=points, facecolors=colors, transOffset=trans,
        edgecolors='none', zorder=1
        )
    col.set_visible(vis)

    treeplot.add_collection(col)
    treeplot.figure.canvas.draw_idle()
Пример #2
0
 def shape(self, height, yrange, rotated):
     g = rlg2mpl.Group()
     (X, Y, I) = (0, 1, 2)
     #scaled_axes = [[X, I], [I, Y]][rotated]
     scaled_axes = [[X, X], [Y, Y]][rotated]
     scaled_axes = [[X, Y], [X, Y]][rotated]
     trans = TransformScalePart(g.combined_transform, scaled_axes)
     a = CircleCollection([.5], edgecolors=self.cvalues,
             facecolors=self.cvalues, offsets=self.offsets,
             transOffset=g.combined_transform)
     g.add(a)
     a.set_transform(trans)
     return g