It's Scanpy friendly!
Please remember to cite us
A Python package for spectral clustering based on the
powerful suite of tools named
too-many-cells.
Note, this is not a wrapper.
In essence, you can use toomanycells to partition a data set
in the form of a matrix of integers or floating point numbers
into clusters, where members of a cluster are similar to each
other under a given similarity function.
The rows represent observations and the
columns are the features. However, sometimes just knowing the
clusters is not sufficient. Often, we are interested on the
relationships between the clusters, and this tool can help
you visualize the clusters as leaf nodes of a tree, where the
branches illustrate the trajectories that have to be followed
to reach a particular cluster. Initially, this tool will
partition your data set into two subsets (each subset is a
node of the tree), trying to maximize
the differences between the two. Subsequently, it will
reapply that same criterion to each subset (node) and will
continue bifurcating until the
modularity
of the node that is about to be partitioned becomes less
than a given threshold value (
- Free software: GNU AFFERO GENERAL PUBLIC LICENSE
- Documentation: https://JRR3.github.io/toomanycells
Version 1.0.40 no longer requires Graphviz. Thus, no need to install a separate C library!
To have control of your working environment you can use a python virtual environment, which can help you keep only the packages you need in one location. In bash or zsh you can simply type
python -m venv /path/to/new/virtual/environment
To activate it you simply need
source pathToTheVirtualEnvironment/bin/activate
To deactivate the environment use the intuitive
deactivate
Caveat: I have tested the following steps in Python 3.9.18. For other versions, things might be different.
In theory, just typing
pip install toomanycells
in your home or custom environment should work. However, for reproducibility, here is the list of packages I had in my virtual environment in 2024:
anndata==0.10.9
array_api_compat==1.8
celltypist==1.6.3
certifi==2024.8.30
charset-normalizer==3.3.2
click==8.1.7
contourpy==1.3.0
cycler==0.12.1
et-xmlfile==1.1.0
exceptiongroup==1.2.2
fonttools==4.54.1
get-annotations==0.1.2
h5py==3.11.0
idna==3.10
igraph==0.11.6
importlib_resources==6.4.5
joblib==1.4.2
kiwisolver==1.4.7
legacy-api-wrap==1.4
leidenalg==0.10.2
llvmlite==0.43.0
matplotlib==3.9.2
natsort==8.4.0
networkx==3.2.1
numba==0.60.0
numpy==2.0.2
openpyxl==3.1.5
packaging==24.1
pandas==2.2.3
patsy==0.5.6
pillow==10.4.0
plotly==5.24.1
pynndescent==0.5.13
pyparsing==3.1.4
python-dateutil==2.9.0.post0
pytz==2024.2
requests==2.32.3
scanpy==1.10.3
scikit-learn==1.5.2
scipy==1.13.1
seaborn==0.13.2
session-info==1.0.0
six==1.16.0
statsmodels==0.14.3
stdlib-list==0.10.0
tenacity==9.0.0
texttable==1.7.0
threadpoolctl==3.5.0
toomanycells==1.0.52
tqdm==4.66.5
tzdata==2024.2
umap-learn==0.5.6
urllib3==2.2.3
zipp==3.20.2
If you want to install an updated version, then please use the following approach.
pip install -U --no-deps toomanycells
Note that we are requiring to keep all the dependencies as they are. Otherwise they would get upgraded and that could potentially break the installation.
To install packages based on a list of requirements, i.e., the packages you want installed with the specific version, then use
pip install -r requirements.txt
where requirements.txt
is a list
like the one shown the above block of code.
Make sure you have the latest version. If not, run the previous command again.
If you want to see a concrete example of how to use toomanycells, check out the jupyter notebook demo.
If you are already familiar with toomanycells and want to generate a quick plot (an SVG) of your tree after calling
tmc_obj.run_spectral_clustering()
then use the following call
tmc_obj.store_outputs(
cell_ann_col="name_of_the_column",
plot_tree=True,
)
or
tmc_obj.store_outputs(
cell_ann_col="name_of_the_column",
plot_tree=True,
draw_modularity=False,
draw_node_numbers=False,
)
with the appropriate flags. If you already have the outputs and you just want to plot, then simply call
tmc_obj.easy_plot(
cell_ann_col="name_of_the_column",
)
or
tmc_obj.easy_plot(
cell_ann_col="name_of_the_column",
draw_modularity=False,
draw_node_numbers=False,
)
with the appropriate flags,
where name_of_the_column
is the name of the AnnData.obs
column that contains the cell annotations.
This function will look for the outputs in the
folder that you defined for the output_directory
as shown in step 2.
Note that this function relies on
too-many-cells
(Ă la Haskell). So you need to have it installed. If you work
within the cluster of your organization, maybe it has
already been installed, and you could load it as follows.
module add too-many-cells
Otherwise, I recommend you to install it with Nix.
Sometimes you want to generate a matrix market
file from a set of genes so that you can visualize
them with other tools. The function that will help you
with this is called create_data_for_tmci
.
Just for context,
imagine you are interested in two genes, COL1A1
, and
TCF21
. Moreover, imagine that you also want to include
in your matrix another feature located in the obs
data
frame called total_counts
. Finally, assume that the
column that contains the labels for your cells is called
cell_annotations
. Please use this a template for your
specific needs.
from toomanycells import TooManyCells as tmc
tmc_obj = tmc(A)
tmc_obj.run_spectral_clustering()
tmc_obj.store_outputs(cell_ann_col="cell_annotations")
list_of_genes = []
list_of_genes.append("COL1A1")
list_of_genes.append("TCF21")
list_of_genes.append("total_counts")
tmc_obj.create_data_for_tmci(list_of_genes=list_of_genes)
These lines of code will produce for you
a folder named tmci_mtx_data
with the expected outputs.
-
First import the module as follows
from toomanycells import TooManyCells as tmc
-
If you already have an AnnData object
A
loaded into memory, then you can create a TooManyCells object withtmc_obj = tmc(A)
In this case the output folder will be called
tmc_outputs
. However, if you want the output folder to be a particular directory, then you can specify the path as follows.tmc_obj = tmc(A, output_directory)
-
If instead of providing an AnnData object you want to provide the directory where your data is located, you can use the syntax
tmc_obj = tmc(input_directory, output_directory)
-
If your input directory has a file in the matrix market format, then you have to specify this information by using the following flag
tmc_obj = tmc(input_directory, output_directory, input_is_matrix_market=True)
Under this scenario, the input_directory
must contain a
.mtx
file, a barcodes.tsv
file (the observations), and
a genes.tsv
(the features).
- Once your data has been loaded successfully,
you can start the clustering process with the following command
tmc_obj.run_spectral_clustering()
In my desktop computer processing a data set with ~90K
cells (observations) and ~30K genes (features) took a
little less than 6 minutes in 1809 iterations. For a
larger data set like the Tabula
Sapiens
with 483,152 cells and 58,870 genes (14.51 GB in zip
format) the total time was about 50 minutes in the same
computer.
- At the end of the clustering process the
.obs
data frame of the AnnData object should have two columns named['sp_cluster', 'sp_path']
which contain the cluster labels and the path from the root node to the leaf node, respectively.tmc_obj.A.obs[['sp_cluster', 'sp_path']]
- To generate the outputs, just call the function
or
tmc_obj.store_outputs()
to include a plot of the graph.tmc_obj.store_outputs( cell_ann_col="name_of_the_column", plot_tree=True, )
This call will generate JSON file
containing the nodes and edges of the graph (graph.json
),
one CSV file that describes the cluster
information (clusters.csv
), another CSV file containing
the information of each node (node_info.csv
), and two
JSON files. One relates cells to clusters
(cluster_list.json
), and the other has the
full tree structure (cluster_tree.json
). You need this
last file for too-many-cells interactive (TMCI).
- If you already have the
graph.json
file you can load it withtmc_obj.load_graph(json_fname="some_path")
- If you want to visualize your results in a dynamic
platform, I strongly recommend the tool
too-many-cells-interactive.
To use it, first make sure that you have Docker Compose and
Docker. One simple way of getting the two is by installing
Docker Desktop.
Note that with MacOS the instructions are slightly different.
If you use Nix, simply
add the packages
pkgs.docker
andpkgs.docker-compose
to your configuration orhome.nix
file and run
home-manager switch
- If you installed Docker Desktop you probably don't need to follow this step. However, under some distributions the following two commands have proven to be essential. Use
sudo dockerd
to start the daemon service for docker containers and
sudo chmod 666 /var/run/docker.sock
to let Docker read and write to that location.
- Now clone the repository
git clone https://github.com/schwartzlab-methods/too-many-cells-interactive.git
and store the path to the too-many-cells-interactive
folder in a variable, for example
path_to_tmc_interactive
. Also, you will need to identify
a column in your AnnData.obs
data frame that has the
labels for the cells. Let's assume that the column name is
stored in the variable cell_annotations
. Lastly, you can
provide a port number to host your visualization, for
instance port_id=1234
. Then, you can call the function
tmc_obj.visualize_with_tmc_interactive(
path_to_tmc_interactive,
cell_annotations,
port_id)
The following visualization corresponds to the data set
with ~90K cells (observations).
And this is the visualization for the Tabula Sapiens data set with ~480K cells.
In case you already have the outputs for TMCI, but you want to visualize a specific set of genes on top of your tree, you are going to need the expression matrix corresponding to those genes in the matrix marker format. You will also need a list of genes and the barcodes. All of that can be easily achieved with toomanycells (Ă la Python) after loading your matrix or AnnData object. If you are interested in only a few genes, you can call
tmc_obj.create_data_for_tmci(
list_of_genes = ["G1","G2",...,"Gn"]
)
where G1
,G2
,...,Gn
, are the labels
of the genes of interest. If instead you have
a table of genes stored as a text file, then
use the call
tmc_obj.create_data_for_tmci(
path_to_genes = "path/to/genes.csv"
)
Lastly, if you want to write all the available genes to a matrix, then simply call
tmc_obj.create_data_for_tmci()
but note that this could take a considerable
amount of time, depending on how many genes
are in your matrix.
After calling this function, you will
have a new folder called tmci_mtx_data
which will contain the aforementioned files.
It is also important to mention that you need
a file wiht the labels
./start-and-load.sh \
--matrix-dir /path_to/tmci_mtx_data \
--tree-path /path_to/cluster_tree.json \
--label-path /path_to/cell_annotations.csv \
--port 2025 \
--debug
To answer that question we have created the following
benchmark. We tested the performance of toomanycells in 20
data sets having the following number of cells: 6,360, 10,479,
12,751, 16,363, 23,973, 32,735, 35,442, 40,784, 48,410, 53,046,
57,621, 62,941, 68,885, 76,019, 81,449, 87,833, 94,543, 101,234,
107,809, and 483,152. The range goes from thousands of cells to
almost half a million cells.
These are the results.
As you can see, the program behaves linearly with respect to the size of the input. In other words, the observations fit the model
When visualizing the tree, we often are interested on
observing how different cell types distribute across the
branches of the tree. In case your AnnData object lacks
a cell annotation column in the obs
data frame, or
if you already have one but you want to try a different
method, we have created a wrapper function that calls
CellTypist. Simply
write
tmc_obj.annotate_with_celltypist(
column_label_for_cell_annotations,
)
and the obs
data frame of your AnnData object will
have a column named like the string stored under the
column_label_for_cell_annotations
variable.
By default we use the Immune_All_High
celltypist
model that contains 32 cell types. If you want to use
another model, simply write
tmc_obj.annotate_with_celltypist(
column_label_for_cell_annotations,
celltypist_model,
)
where celltypist_model
describes the type of model
to use by the library. For example, if this
variable is equal to Immune_All_Low
, then the number
of possible cell types increases to 98.
For a complete list of all the models, see the following
list. Lastly,
if you want to use the fact that transcriptionally similar
cells are likely to cluster together, you can assign the cell
type labels on a cluster-by-cluster basis
rather than a cell-by-cell basis. To activate this
feature, use the call
tmc_obj.annotate_with_celltypist(
column_label_for_cell_annotations,
celltypist_model,
use_majority_voting = True,
)
If you want to select cells
that belong to a class defined
within a specific column of the
.obs
dataframe, you can use the
following call.
A = tmc_obj.filter_for_cells_with_property(
"cell_type", "Neuro-2a")
In this case all cells that have the label Neuro-2a
within the column cell_type
in the .obs
dataframe
will be selected, and the resulting AnnData object A
will only have these cells.
If you use MEDUSA
, please refer to
this YAML file
to have an idea of how you can use some of
the following functions from TooManyCells Ă la Python.
Imagine you have a tree structure
of your data like the one shown below.
If you want to isolate all the cells that belong to branches
261 and 2, and produce an AnnData object with those cells,
simply use the following call
adata = tmc_obj.isolate_cells_from_branches(
list_of_branches=[261,2])
If you have a CSV file that specifies the branches, then use the following call
adata = tmc_obj.isolate_cells_from_branches(
path_to_csv_file="list_of_branches.csv",
branch_column="node",
)
The name of the column that contains the branches
or nodes is specified through the keyword
branch_column
. Lastly, if you want to store
a copy of the indices, use the following call
adata = tmc_obj.isolate_cells_from_branches(
path_to_csv_file="list_of_branches.csv",
branch_column="node",
generate_cell_id_file=True,
)
Imagine we have the following tree.
If you want to quantify the mean expression of the marker
CD9 on branch 261, you can use the following call
m_exp = tmc_obj.compute_cluster_mean_expression(
node=261, genes=["CD9"])
and you would obtain 12.791.
Looking at the above plot, this suggests that Neuro-2a cells highly express this marker. If instead we were interested in a different marker, like SDC1, this would be the corresponding color map expression across the nodes.
The above plot also illustrates that some Neuro-2a cells are rich in SDC1.
First we introduce the concept of median absolute
deviation. Imagine you have a list of
Based on the previous example, now imagine you want to find cells whose expression of two markers, CD9 and SDC1, is 1 MAD above the median. First, you need a CSV file containing the following information.
Marker Cell Threshold Direction
CD9 Neuro-2a 1.0 Above
SDC1 Neuro-2a 1.0 Above
Let's call it marker_and_cell_info.csv
.
Note: For this discussion the cell types indicated in
the Cell
column are not relevant and will not be
used. We quantify the mean expression of those
markers for every node of the tree and store
that information within each node.
We can do that using the following call.
tmc_obj.populate_tree_with_mean_expression_for_all_markers(
cell_marker_path="marker_and_cell_info.csv")
Then we compute basic statistics for each marker using the following function
tmc_obj.compute_node_expression_metadata()
These are the statistics associated to those markers.
median mad min max min_mad max_mad delta
CD9 3.080538 1.918258 0.000890 22.944445 -1.605441 10.355182 0.797375
SDC1 2.989691 1.165005 0.001669 6.639456 -2.564814 3.132832 0.379843
Note that the maximum expression of CD9 is about 10 MADs above the median, while that of SDC1 is only about 3 MADs above the median. The plot corresponding to the distribution of those markers across all nodes can be generated through this call
tmc_obj.plot_marker_distributions()
The plots will be all contained in a dynamic html file. Here are some examples.
This is the distribution for CD9:
and with TooManyCellsInteractive
The distribution for SDC1 looks as follows.
If we want to isolate the cells that satisfy the conditions
Marker Cell Threshold Direction
CD9 Neuro-2a 1.0 Above
SDC1 Neuro-2a 1.0 Above
We can use the call
tmc_obj.select_cells_based_on_inequalities(
cell_ann_col="cell_type")
where the cell annotation column in the .obs
dataframe is specified through the
cell_ann_col
keyword. This function will
return an AnnData object with all the
cells satisfying all the constraints.
This function will also produce
multiple CSV files. One for each inequality
specified through the file of constraints.
For example,
one for all cells whose
expression of CD9 was above 1 MAD of the median
expression of CD9,
one for all cells whose
expression of SDC1 was above 1 MAD of the median
expression of SDC1,
and one corresponding to the intersection
of all of the above. The above function will
modify the original AnnData object by adding to
the .obs
dataframe a column
named Intersection
indicating with a boolean value
if a cell satisfies all the constraints.
Lastly, if the number of markers is less than or
equal to three, then the .obs
dataframe will
include a column classifying the cells
based on whether they express
highly or not each of the markers. For instance,
in this example we obtained the following outputs.
Class
CD9-Low-SDC1-Low 28729
CD9-High-SDC1-Low 6072
CD9-High-SDC1-High 4223
CD9-Low-SDC1-High 2058
Name: count, dtype: int64
Class
CD9-Low-SDC1-Low 0.699309
CD9-High-SDC1-Low 0.147802
CD9-High-SDC1-High 0.102794
CD9-Low-SDC1-High 0.050095
Name: proportion, dtype: float64
This indicates that the majority of the cells,
i.e., about 70%
of cells,
are low in CD9 and low in SDC1, and about 10%
of cells
are high in both. Note that in this particular example
when we say high it means
that the expression is above 1 MAD from the median, and
low is the complement of that.
Imagine you want to compare the heterogeneity of cell
populations belonging to different branches of the
toomanycells tree. By branch we mean all the nodes that
derive from a particular node, including the node
that defines the branch in question.
For example, we want to compare branch 1183 against branch 2.
One way to do this is by comparing the modularity
distribution and the cumulative modularity for all the
nodes that belong to each branch.
We can do that using the following calls. First for
branch 1183
tmc_obj.quantify_heterogeneity(
list_of_branches=[1183],
use_log_y=true,
tag="branch_A",
show_column_totals=true,
color="blue",
file_format="svg")
And then for branch 2
tmc_obj.quantify_heterogeneity(
list_of_branches=[2],
use_log_y=true,
tag="branch_B",
show_column_totals=true,
color="red",
file_format="svg")
Note that you can include multiple nodes in the list of branches. From these figures we observe that the higher cumulative modularity of branch 1183 with respect to branch 2 suggests that the former has a higher degree of heterogeneity. However, just relying in modularity could provide a misleading interpretation. For example, consider the following scenario where the numbers within the nodes indicate the modularity at that node.
In this case, scenario A has a larger cumulative modularity, but we note that scenario B is more heterogeneous. For that reason we recommend also computing additional diversity measures. First, we need some notation. For all the branches belonging to the list of branches in the above function
quantify_heterogeneity
, let
then we define the following diversity measure
In general, the larger the value of
When
which represents the
probability that two cells picked at random belong
to the same species. Hence, the higher the Simpson's
index, the less diverse is the ecosystem.
Lastly, when
In the above example, for branch 1183 we obtain
value
Richness 460.000000
Shannon 5.887544
Simpson 0.003361
MaxProp 0.010369
q = 0 460.000000
q = 1 360.518784
q = 2 297.562094
q = inf 96.442786
and for branch 2 we obtain
value
Richness 280.000000
Shannon 5.500414
Simpson 0.004519
MaxProp 0.010750
q = 0 280.000000
q = 1 244.793371
q = 2 221.270778
q = inf 93.021531
After comparing the results using two different measures, namely, modularity and diversity, we conclude that branch 1183 is more heterogeneous than branch 2.
So far we have assumed that the similarity matrix
However, this is not the only way to compute a similarity matrix. We will list all the available similarity functions and how to call them.
If your matrix is sparse, i.e., the number of nonzero
entries is proportional to the number of samples (
tmc_obj.run_spectral_clustering(
similarity_function="cosine_sparse")
By default we use the ARPACK library (written in Fortran) to compute the truncated singular value decomposition. The Halko-Martinsson-Tropp algorithm is also available. However, this one is not deterministic.
tmc_obj.run_spectral_clustering(
similarity_function="cosine_sparse",
svd_algorithm="arpack")
If
If your data consists of points whose Euclidean norm varies across multiple length scales, then one option is to use a similarity function that can adapt to those changes in magnitude. Before I explain it in detail, here is how you can call it
tmc_obj.run_spectral_clustering(
similarity_function="norm_sparse")
If your matrix is dense, and you want to use the cosine similarity, then use the following instruction.
tmc_obj.run_spectral_clustering(
similarity_function="cosine")
The same comment about negative entries applies here.
However, there is a simple solution. While shifting
the matrix of observations can drastically change the
interpretation of the data because each column lives
in a different (gene) space, shifting the similarity
matrix is actually a reasonable method to remove negative
entries. The reason is that similarities live in an
ordered space and shifting by a constant is
an order-preserving transformation. Equivalently,
if the similarity between
tmc_obj.run_spectral_clustering(
similarity_function="cosine",
shift_similarity_matrix=1)
Note that since the range of the cosine similarity
is
The similarity matrix is given by
This is an example:
tmc_obj.run_spectral_clustering(
similarity_function="laplacian",
similarity_gamma=0.01)
This function is very sensitive to
a smaller value for
The similarity matrix is given by
This is an example:
tmc_obj.run_spectral_clustering(
similarity_function="gaussian",
similarity_gamma=0.001)
As before, this function is very sensitive to
The similarity matrix is given by
where
tmc_obj.run_spectral_clustering(
similarity_function="div_by_sum")
If you want to use the inverse document frequency (IDF) normalization, then use
tmc_obj.run_spectral_clustering(
similarity_function="some_sim_function",
use_tf_idf=True)
If you also want to normalize the frequencies to
unit norm with the
tmc_obj.run_spectral_clustering(
similarity_function="some_sim_function",
use_tf_idf=True,
tf_idf_norm="l2")
If instead you want to use the
Sometimes normalizing your matrix of observations can improve the performance of some routines. To normalize the rows, use the following instruction.
tmc_obj.run_spectral_clustering(
similarity_function="some_sim_function",
normalize_rows=True)
Be default, the
tmc_obj.run_spectral_clustering(
similarity_function="some_sim_function",
normalize_rows=True,
similarity_norm=p)
Imagine you have the following tree structure after
running toomanycells.
Further, assume that the colors denote different classes
satisfying specific properties. We want to know how the
expression of two genes, for instance,
Gene S
and Gene T
,
fluctuates as we move from node Class B
, to node Class C
. To compute such quantities, we first need to define the
distance between nodes.
Assume we have a (parent) node
We also define
as expected. Now that we know how to calculate the
distance between a node and its parent or child, let
-
$N_0 = X$ , -
$N_n=Y$ , -
$N_i$ is a direct relative of$N_{i+1}$ , i.e.,$N_i$ is either a child or parent of$N_{i+1}$ , -
$N_i \neq N_j$ for$i\neq j$ .
Then, the distance between
We define the expression
of Gene G
at a node Gene G
considering all the cells that belong to node
we can compute the corresponding gene expression sequence
Lastly, since we are interested in plotting the
gene expression as a function of the distance with respect to
the node
- The sequence of nodes between
$X$ and$Y$ $${(N_{i})}_{i=0}^{n}$$ - The sequence of gene expression levels between
$X$ and$Y$ $${(E_{i})}_{i=0}^{n}$$ - And the sequence of distances with respect to node
$X$ $${(D_{i})}_{i=0}^{n}$$
The final plot is simply
Note how the expression of Gene A
is high relative to
that of Gene B
at node Gene B
is
highly expressed relative to Gene A
at node
I would like to thank the Schwartz lab (GW) for letting me explore different directions and also Christie Lau for providing multiple test cases to improve this implementation.