sciv 0.0.93__py3-none-any.whl → 0.0.95__py3-none-any.whl

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)
 

{sciv-0.0.93.dist-info → sciv-0.0.95.dist-info}/METADATA

@@ -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.dist-info → sciv-0.0.95.dist-info}/RECORD

@@ -16,7 +16,7 @@ sciv/plot/_kde_.py,sha256=9VC6DKHbQAzKenz4DoiL5LgUzEJBtAWXXPCLxOV74kY,2459
 sciv/plot/_line_.py,sha256=P944YoGxSocHncjtme4n_iIPK9sghNW3-Xee6Tyy4HE,3850
 sciv/plot/_pie_.py,sha256=OVPv85MdjkjABh-uP5Y-KVk1VygFIzz87QT3OE1hZjU,6399
 sciv/plot/_radar_.py,sha256=g_abzmzibJIR6it59TendUI236inbzYl0IzzdoA3Uuc,6304
-sciv/plot/_scatter_.py,sha256=ZRP2-nRMygrZtj6tQlOqp8n5_FsWRe9cx9d80DhoezQ,16708
+sciv/plot/_scatter_.py,sha256=HPy1eFqDk4vksAiKjn_Qk00fYO4GyqRs1O3ZZh44Ezw,16998
 sciv/plot/_venn_.py,sha256=TfNTuxog2pT7sicKBEEMtleoHXwnenbl3CMWJu9c2vs,2675
 sciv/plot/_violin_.py,sha256=40LYeHFYyNL1XbKHTjmwLT0zzh7o5emqN9Gl2sbl-DA,6458
 sciv/preprocessing/__init__.py,sha256=56RgDai5I3sZ4hl3aaV80ogOeUscYsU3nJUWE80jZ-k,742
@@ -33,7 +33,7 @@ sciv/tool/_random_walk_.py,sha256=lnMiJyDuqDB75l9IkFLr7bIgKYhWxlBdIcExSErpQNw,48
 sciv/util/__init__.py,sha256=nOxZ8if27X7AUJ6hZwTwxOJwIBJb0obWlHjqCzjg_Gc,1964
 sciv/util/_constant_.py,sha256=w0wKQd8guLd1ZTW24_5aECrWsIWDiNQmEpLsWlHar1A,3000
 sciv/util/_core_.py,sha256=hF33ybPcoVlapZsm-2Etem-p_rQUqXlsdaQgZv5jD7w,14867
-sciv-0.0.93.dist-info/METADATA,sha256=CBRawJaX69GIAtLlTBO_dylZ60jF1Z8euWcDEcQNdYU,3465
-sciv-0.0.93.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-sciv-0.0.93.dist-info/licenses/LICENSE,sha256=4UvHVf3qCOZjHLs4LkYz8u96XRpXnZrpTKrkUQPs5_A,1075
-sciv-0.0.93.dist-info/RECORD,,
+sciv-0.0.95.dist-info/METADATA,sha256=WoU9-ogUBNJP28XtO8OS4GsNzGwjkqdDZqV8gXihBB0,3465
+sciv-0.0.95.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+sciv-0.0.95.dist-info/licenses/LICENSE,sha256=4UvHVf3qCOZjHLs4LkYz8u96XRpXnZrpTKrkUQPs5_A,1075
+sciv-0.0.95.dist-info/RECORD,,