SURE-tools 2.1.45__py3-none-any.whl → 2.1.47__py3-none-any.whl

SURE/flow/flow_stats.py

@@ -1,4 +1,5 @@
 import numpy as np
+from scipy.interpolate import griddata
 from scipy.spatial.distance import pdist, squareform
 from sklearn.decomposition import PCA
 from scipy.stats import pearsonr
@@ -16,6 +17,30 @@ class VectorFieldEval:
         
     def momentum_flow_metric(self, vectors, masses=None):
         return momentum_flow_metric(vectors=vectors, masses=masses)
+    
+    def divergence(self, points, vectors, grid_resolution=30):
+        # 提取坐标和向量分量
+        x_coords = points[:, 0]
+        y_coords = points[:, 1]
+        u_components = vectors[:, 0]  # x方向分量
+        v_components = vectors[:, 1]  # y方向分量
+    
+        # 创建规则网格
+        x_grid = np.linspace(x_coords.min(), x_coords.max(), grid_resolution)
+        y_grid = np.linspace(y_coords.min(), y_coords.max(), grid_resolution)
+        X, Y = np.meshgrid(x_grid, y_grid)
+    
+        # 插值到网格
+        U_grid = griddata((x_coords, y_coords), u_components, (X, Y), method='linear')
+        V_grid = griddata((x_coords, y_coords), v_components, (X, Y), method='linear')
+    
+        # 计算散度
+        dU_dx = np.gradient(U_grid, x_grid, axis=1)
+        dV_dy = np.gradient(V_grid, y_grid, axis=0)
+        divergence = dU_dx + dV_dy
+        divergence[np.isnan(divergence)] = 0
+        
+        return divergence
 
         
 def calculate_movement_stats(vectors):

SURE/perturb/perturb.py

@@ -9,7 +9,7 @@ class LabelMatrix:
     def __init__(self):
         self.labels_ = None
     
-    def fit_transform(self, labels, control_label=None, sep_pattern=r'[;_\-\s]', speedup: Literal['none','vectorize','parallel']='none'):
+    def fit_transform(self, labels, control_label=None, sep_pattern=r'[,;_\s]', speedup: Literal['none','vectorize','parallel']='none'):
         if speedup=='none':
             mat, self.labels_ = label_to_matrix(labels=labels, sep_pattern=sep_pattern)
         elif speedup=='vectorize':

{sure_tools-2.1.45.dist-info → sure_tools-2.1.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 2.1.45
+Version: 2.1.47
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-2.1.45.dist-info → sure_tools-2.1.47.dist-info}/RECORD

@@ -9,17 +9,17 @@ SURE/atac/utils.py,sha256=m4NYwpy9O5T1pXTzgCOCcmlwrC6GTi-cQ5sm2wZu2O8,4354
 SURE/codebook/__init__.py,sha256=2T5gjp8JIaBayrXAnOJYSebQHsWprOs87difpR1OPNw,243
 SURE/codebook/codebook.py,sha256=ZlN6gRX9Gj2D2u3P5KeOsbZri0MoMAiJo9lNeL-MK-I,17117
 SURE/flow/__init__.py,sha256=rsAjYsh1xVIrxBCuwOE0Q_6N5th1wBgjJceV0ABPG3c,183
-SURE/flow/flow_stats.py,sha256=cBBsPEDpWNMpbzlyQ3f0385RSrX6_5RCH2caOyi4ihM,9908
+SURE/flow/flow_stats.py,sha256=_pF7m4-87SKlCHVtVmx3LG2bAGVXOnAfEgMzLhLx4Io,10910
 SURE/flow/plot_quiver.py,sha256=UbmuScUcgbQHeMmjKmgqxjrIjHhiHx0VWct16UMMwuE,8110
 SURE/perturb/__init__.py,sha256=8TP1dSUhXiZzKpFebHZmm8XMMGbUz_OfQ10xu-6uPPY,43
-SURE/perturb/perturb.py,sha256=ICs-yMIuT0tZtyx9R7tkzzhAcurs_eActv2v3uyjreE,5038
+SURE/perturb/perturb.py,sha256=1iSsCePcwkA2CyM1nCdq_G8gogUNjhMH0BfhhvhpJQk,5037
 SURE/utils/__init__.py,sha256=QJUOfrXzdWSmoM0P3LH8oKEHttzCWqpDy2UF0F0dtN4,673
 SURE/utils/custom_mlp.py,sha256=rHnx9jEef02zfCUdbYVCmbuHcDdIBmRgt__wpdpZvYg,8104
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.1.45.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.1.45.dist-info/METADATA,sha256=y0VumQ_zuHF_45Mo2MKKap2J_l4rGEzHiYLSxuq1isw,2651
-sure_tools-2.1.45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.1.45.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
-sure_tools-2.1.45.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.1.45.dist-info/RECORD,,
+sure_tools-2.1.47.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.1.47.dist-info/METADATA,sha256=wysn-KRqcUVbYerQ9FK7D6LLDC4b5KB_IYzTw0uJzVw,2651
+sure_tools-2.1.47.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.1.47.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
+sure_tools-2.1.47.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.1.47.dist-info/RECORD,,