tomodrgn.commands.pc_traversal.analyze_data_support

analyze_data_support(z: ndarray, traj: ndarray, cutoff: int = 3) → ndarray

Count the number of neighbors in reference array z within Euclidean distance cutoff of query points traj.

Parameters:

• z – array of latent embeddings, shape (nptcls, zdim)

• traj – trajectory of points through latent embedding space, shape (args.n, zdim)

• cutoff – Euclidean distance within which to consider points in z as neighbors of points in traj

Returns:

array of neighbor counts at each point in traj, shape (args.n)