Получите ограничительные рамки отдельных элементов PathCollection из plt.scatter

Question

есть ли способ получить ограничивающие рамки отдельных элементов в выводе plt.scatter ()? Я могу получить смещения (то есть координаты x и y - с которых я должен начать, так как я использую их для построения графиков) и размеры, но размеры не в единицах данных, так что даже случайное преобразование из области в радиус для получения размеров bboxes не будет работать ...

Есть ли хороший способ сделать это?

tips = sns.load_dataset('tips')[:20]
f, ax = plt.subplots()
sc = ax.scatter(tips["total_bill"], y=tips["tip"], s=(tips["size"]*3)**2)
plt.show()

sc.properties()['offsets']

array([[ 16.99,   1.01],
       [ 10.34,   1.66],
       [ 21.01,   3.5 ],
       [ 23.68,   3.31],
       [ 24.59,   3.61],
       [ 25.29,   4.71],
       [  8.77,   2.  ],
       [ 26.88,   3.12],
       [ 15.04,   1.96],
       [ 14.78,   3.23],
       [ 10.27,   1.71],
       [ 35.26,   5.  ],
       [ 15.42,   1.57],
       [ 18.43,   3.  ],
       [ 14.83,   3.02],
       [ 21.58,   3.92],
       [ 10.33,   1.67],
       [ 16.29,   3.71],
       [ 16.97,   3.5 ],
       [ 20.65,   3.35]])

sc.get_sizes()

array([ 36, 81, 81, 36, 144, 144, 36, 144, 36, 36, 36, 144, 36, 144, 36, 36, 81, 81, 81, 81])

Answer 1

В целом это далеко не просто. PathCollection допускает различные преобразования, а также преобразования смещения. Также он может иметь один или несколько путей и размеров.

К счастью, есть встроенная функция matplotlib.path.get_path_collection_extents, которая предоставляет ограничивающий прямоугольник PathCollection. Мы можем использовать это, чтобы вместо этого получить экстент каждого отдельного члена, предоставив список из одного элемента каждого отдельного пути и пройдя по всем из них.
Поскольку ограничивающая рамка указана в пикселях, необходимо в конце преобразовать обратно в координаты данных.

Ниже приведена полная функция, которая сделает все это. Сначала нужно будет нарисовать фигуру, чтобы были установлены различные преобразования.

import numpy as np; np.random.seed(432)
import matplotlib.pyplot as plt
from matplotlib.path import get_path_collection_extents


def getbb(sc, ax):
    """ Function to return a list of bounding boxes in data coordinates
        for a scatter plot """
    ax.figure.canvas.draw() # need to draw before the transforms are set.
    transform = sc.get_transform()
    transOffset = sc.get_offset_transform()
    offsets = sc._offsets
    paths = sc.get_paths()
    transforms = sc.get_transforms()

    if not transform.is_affine:
        paths = [transform.transform_path_non_affine(p) for p in paths]
        transform = transform.get_affine()
    if not transOffset.is_affine:
        offsets = transOffset.transform_non_affine(offsets)
        transOffset = transOffset.get_affine()

    if isinstance(offsets, np.ma.MaskedArray):
        offsets = offsets.filled(np.nan)

    bboxes = []

    if len(paths) and len(offsets):
        if len(paths) < len(offsets):
            # for usual scatters you have one path, but several offsets
            paths = [paths[0]]*len(offsets)
        if len(transforms) < len(offsets):
            # often you may have a single scatter size, but several offsets
            transforms = [transforms[0]]*len(offsets)

        for p, o, t in zip(paths, offsets, transforms):
            result = get_path_collection_extents(
                transform.frozen(), [p], [t],
                [o], transOffset.frozen())
            bboxes.append(result.inverse_transformed(ax.transData))

    return bboxes



fig, ax = plt.subplots()

sc = ax.scatter(*np.random.rand(2,5), s=np.random.rand(5)*150+60)

# a single size needs to work as well. As well as a marker with non-square extent
sc2 = ax.scatter([0.2,0.5],[0.1, 0.7], s=990, marker="$\\rightarrow$")  

boxes = getbb(sc, ax)
boxes2 = getbb(sc2, ax)

# Draw little rectangles for boxes:
for box in boxes+boxes2:
    rec = plt.Rectangle((box.x0, box.y0), box.width, box.height, fill=False,
                        edgecolor="crimson")
    ax.add_patch(rec)

plt.show()