Source code for data_morph.shapes.lines.rectangle
"""Rectangle shape."""
from ...data.dataset import Dataset
from ..bases.line_collection import LineCollection
[docs]
class Rectangle(LineCollection):
"""
Class for the rectangle shape.
.. plot::
:scale: 75
:caption:
This shape is generated using the panda dataset.
import matplotlib.pyplot as plt
from data_morph.data.loader import DataLoader
from data_morph.shapes.polygons import Rectangle
_ = Rectangle(DataLoader.load_dataset('panda')).plot()
Parameters
----------
dataset : Dataset
The starting dataset to morph into other shapes.
"""
def __init__(self, dataset: Dataset) -> None:
xmin, xmax = dataset.df.x.quantile([0.1, 0.9])
ymin, ymax = dataset.df.y.quantile([0.1, 0.9])
super().__init__(
[[xmin, ymin], [xmin, ymax]],
[[xmin, ymin], [xmax, ymin]],
[[xmax, ymin], [xmax, ymax]],
[[xmin, ymax], [xmax, ymax]],
)
