sciv 0.0.93__tar.gz → 0.0.95__tar.gz

{sciv-0.0.93 → sciv-0.0.95}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sciv
-Version: 0.0.93
+Version: 0.0.95
 Summary: Unveiling the pivotal cell types involved in variant function regulation at a single-cell resolution
 Project-URL: github, https://github.com/YuZhengM/sciv
 Author-email: Zheng-Min Yu <yuzmbio@163.com>

{sciv-0.0.93 → sciv-0.0.95}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 [project]
 name = "sciv"
-version = "0.0.93"
+version = "0.0.95"
 authors = [
     { name = "Zheng-Min Yu", email = "yuzmbio@163.com" },
 ]

{sciv-0.0.93 → sciv-0.0.95}/src/sciv/plot/_scatter_.py

@@ -8,6 +8,7 @@ import numpy as np
 import pandas as pd
 from anndata import AnnData
 from matplotlib import pyplot as plt
+from matplotlib.colors import ListedColormap
 from pandas import DataFrame
 import seaborn as sns
 
@@ -163,11 +164,15 @@ def scatter_3d(
     y_name: str = None,
     z_name: str = None,
     title: str = None,
-    width: float = 2,
-    height: float = 2,
-    bottom: float = 0,
+    width: float = 7,
+    height: float = 7,
+    elev: float = 30,
+    azim: float = -60,
+    is_add_legend: bool = True,
+    cmap: Union[str, ListedColormap] = 'tab20',
+    font_size: int = 14,
     edge_color: str = None,
-    size: Union[float, collection] = 1.0,
+    size: Union[float, collection] = 0.1,
     legend_name: str = None,
     output: path = None,
     show: bool = True,
@@ -178,8 +183,8 @@ def scatter_3d(
         ul.log(__name__).error(f"At least one of the `output` and `show` parameters is required")
         raise ValueError(f"At least one of the `output` and `show` parameters is required")
 
-    fig, ax = plt.subplots(figsize=(width, height))
-    fig.subplots_adjust(bottom=bottom, projection='3d')
+    fig = plt.figure(figsize=(width, height))
+    ax = fig.add_subplot(projection='3d')
 
     hue_cat = pd.Categorical(df[hue])
 
@@ -188,36 +193,38 @@ def scatter_3d(
         df[y],
         df[z],
         c=hue_cat.codes,
-        cmap='tab20',
+        cmap=cmap,
         s=size,
         edgecolors=edge_color,
         **kwargs
     )
 
-    # 设置坐标轴标签和标题
+    # angle of view
+    ax.view_init(elev=elev, azim=azim)
 
     if x_name is not None:
-        ax.set_xlabel(x_name)
+        ax.set_xlabel(x_name, fontsize=font_size)
 
     if y_name is not None:
-        ax.set_ylabel(y_name)
+        ax.set_ylabel(y_name, fontsize=font_size)
 
     if z_name is not None:
-        ax.set_zlabel(z_name)
+        ax.set_zlabel(z_name, fontsize=font_size)
 
     if title is not None:
-        ax.set_title(title)
-
-    unique_types = hue_cat.categories
-    legend_elements = [
-        plt.Line2D(
-            [0], [0], marker='o', color='w', label=type_,
-            markerfacecolor=scatter.cmap(scatter.norm(i))
-        )
-        for i, type_ in enumerate(unique_types)
-    ]
-
-    ax.legend(handles=legend_elements, title=legend_name, bbox_to_anchor=(1.05, 1), loc='upper left')
+        ax.set_title(title, fontsize=font_size)
+
+    if is_add_legend:
+        unique_types = hue_cat.categories
+        legend_elements = [
+            plt.Line2D(
+                [0], [0], marker='o', color='w', label=type_,
+                markerfacecolor=scatter.cmap(scatter.norm(i))
+            )
+            for i, type_ in enumerate(unique_types)
+        ]
+
+        ax.legend(handles=legend_elements, title=legend_name, loc='upper left')
 
     plot_end(fig, None, None, None, output, show, close)