ddd_subplots's Introduction

DDD Subplots

Python package making it easier to handle mixed 3d and 2d subplots.

How do I install this package?

As usual, just download it using pip:

pip install ddd_subplots

Usage Example

3D subplots

To get a set of 3d subplots just import subplots:

from ddd_subplots import subplots

fig, axes = subplots(1, 3, figsize=(15, 5))

Rotating 3D scatter plots

The library also offers a method to render 3D scatter plots. Here's a complete example:

from ddd_subplots import subplots, rotate
import numpy as np
from sklearn import datasets
from sklearn.decomposition import PCA


def write_frame(X_reduced, y):
    colors = np.array(["red", "green", "blue"])[y]
    fig, axes = subplots(1, 3, figsize=(15, 5))
    for axis in axes.flatten():
        axis.scatter(*X_reduced.T, depthshade=False,
                    c=colors, marker='o', s=20)
    fig.tight_layout()
    return fig, axes

X, y = datasets.load_iris(return_X_y=True)
X_reduced = PCA(n_components=3).fit_transform(X)

rotate(
    write_frame,
    X_reduced,
    "test_animation.gif",
    y,
    duration=10,
    verbose=True
)

Output:

ddd_subplots's People

Contributors

Stargazers

Watchers

ddd_subplots's Issues

Update readme with the new my_func

def my_func(points: np.ndarray, *args, **kwargs):
    fig, axis = subplots(1, 1, figsize=(6, 6), dpi=200)
    s = axis.scatter(*points, **kwargs)
    axis.set_xticklabels([])
    axis.set_yticklabels([])
    axis.set_zticklabels([])
    axis.set_xlim(-0.25, 0.25)
    axis.set_ylim(-0.25, 0.25)
    axis.set_zlim(-0.25, 0.25)
    axis.set_axis_off()
    axis.set_title("3d-TSNE of the selected data colored by cohort.")
    axis.legend(
        handles=s.legend_elements()[0],
        labels=["Core", "Extended"],
        loc="lower right",
        title="Cohort",
    )
    return fig, axis
    ```

Recommend Projects