skbio.diversity.beta.weighted_unifrac¶
- skbio.diversity.beta.weighted_unifrac(u_counts, v_counts, otu_ids, tree, normalized=False, validate=True)[source]¶
Compute weighted UniFrac with or without branch length normalization
State: Experimental as of 0.4.1.
- Parameters:
u_counts (list, np.array) – Vectors of counts/abundances of OTUs for two samples. Must be equal length.
v_counts (list, np.array) – Vectors of counts/abundances of OTUs for two samples. Must be equal length.
otu_ids (list, np.array) – Vector of OTU ids corresponding to tip names in
tree
. Must be the same length asu_counts
andv_counts
.tree (skbio.TreeNode) – Tree relating the OTUs in otu_ids. The set of tip names in the tree can be a superset of
otu_ids
, but not a subset.normalized (boolean, optional) – If
True
, apply branch length normalization, which is described in [1]. Resulting distances will then be in the range[0, 1]
.validate (bool, optional) – If False, validation of the input won’t be performed. This step can be slow, so if validation is run elsewhere it can be disabled here. However, invalid input data can lead to invalid results or error messages that are hard to interpret, so this step should not be bypassed if you’re not certain that your input data are valid. See
skbio.diversity
for the description of what validation entails so you can determine if you can safely disable validation.
- Returns:
The weighted UniFrac distance between the two samples.
- Return type:
- Raises:
ValueError, MissingNodeError, DuplicateNodeError – If validation fails. Exact error will depend on what was invalid.
Notes
Weighted UniFrac was originally described in [1], which includes a discussion of unweighted (qualitative) versus weighted (quantitiative) diversity metrics. Deeper mathemtical discussions of this metric is presented in [2].
If computing weighted UniFrac for multiple pairs of samples, using
skbio.diversity.beta_diversity
will be much faster than calling this function individually on each sample.This implementation differs from that in PyCogent (and therefore QIIME versions less than 2.0.0) by imposing a few additional restrictions on the inputs. First, the input tree must be rooted. In PyCogent, if an unrooted tree was provided that had a single trifurcating node (a newick convention for unrooted trees) that node was considered the root of the tree. Next, all OTU IDs must be tips in the tree. PyCogent would silently ignore OTU IDs that were not present the tree. To reproduce UniFrac results from PyCogent with scikit-bio, ensure that your PyCogent UniFrac calculations are performed on a rooted tree and that all OTU IDs are present in the tree.
This implementation of weighted UniFrac is the array-based implementation described in [3].
If using large number of samples or a large tree, we advise using the optimized UniFrac library [4].
References
Examples
Assume we have the following abundance data for two samples,
u
andv
, represented as a pair of counts vectors. These counts represent the number of times specific Operational Taxonomic Units, or OTUs, were observed in each of the samples.>>> u_counts = [1, 0, 0, 4, 1, 2, 3, 0] >>> v_counts = [0, 1, 1, 6, 0, 1, 0, 0]
Because UniFrac is a phylogenetic diversity metric, we need to know which OTU each count corresponds to, which we’ll provide as
otu_ids
.>>> otu_ids = ['OTU1', 'OTU2', 'OTU3', 'OTU4', 'OTU5', 'OTU6', 'OTU7', ... 'OTU8']
We also need a phylogenetic tree that relates the OTUs to one another.
>>> from io import StringIO >>> from skbio import TreeNode >>> tree = TreeNode.read(StringIO( ... '(((((OTU1:0.5,OTU2:0.5):0.5,OTU3:1.0):1.0):0.0,' ... '(OTU4:0.75,(OTU5:0.5,((OTU6:0.33,OTU7:0.62):0.5' ... ',OTU8:0.5):0.5):0.5):1.25):0.0)root;'))
Compute the weighted UniFrac distance between the samples.
>>> from skbio.diversity.beta import weighted_unifrac >>> wu = weighted_unifrac(u_counts, v_counts, otu_ids, tree) >>> print(round(wu, 2)) 1.54
Compute the weighted UniFrac distance between the samples including branch length normalization so the value falls in the range
[0.0, 1.0]
.>>> wu = weighted_unifrac(u_counts, v_counts, otu_ids, tree, ... normalized=True) >>> print(round(wu, 2)) 0.33