Function rank_gene_groups_dataframe

src/memory/processing/diffexp/mod.rs:243–348

(
    adata: &IMAnnData,
    groupby: &str,
    reference: Option<&str>,
    groups: Option<&[&str]>,
    method: Option<TestMethod>,
    n_genes: Option<usize>,
    correction_method: CorrectionMetho

#[allow(clippy::too_many_arguments)]
pub fn rank_gene_groups_dataframe(
    adata: &IMAnnData,
    groupby: &str,
    reference: Option<&str>,
    groups: Option<&[&str]>,
    method: Option<TestMethod>,
    n_genes: Option<usize>,
    correction_method: CorrectionMethod,
    compute_logfoldchanges: Option<bool>,
    pseudocount: Option<f64>,
) -> anyhow::Result<DataFrame> {
    let method = method.unwrap_or(TestMethod::TTest(TTestType::Welch));
    let compute_lfc = compute_logfoldchanges.unwrap_or(true);
    let pseudocount = pseudocount.unwrap_or(1.0);
    let n_genes = n_genes.unwrap_or(adata.n_vars());

    let computation = compute_rank_gene_groups(
        adata,
        groupby,
        reference,
        groups,
        method,
        n_genes,
        correction_method,
        compute_lfc,
        pseudocount,
    )?;

    let RankGeneGroupsComputed {
        groups: groups_to_test,
        results,
        reference,
    } = computation;

    let DifferentialExpressionResults {
        mut scores,
        mut pvals,
        mut pvals_adj,
        mut logfoldchanges,
        mut gene_names,
    } = results;

    let mut group_column: Vec<String> = Vec::new();
    let mut reference_column: Vec<String> = Vec::new();
    let mut gene_column: Vec<String> = Vec::new();
    let mut score_column: Vec<f64> = Vec::new();
    let mut pval_column: Vec<f64> = Vec::new();
    let mut pval_adj_column: Vec<f64> = Vec::new();
    let mut logfc_column: Vec<f64> = Vec::new();

    let reference_label = reference.as_deref().unwrap_or("rest");

    for group in &groups_to_test {
        let group_scores = scores
            .remove(group)
            .ok_or_else(|| anyhow::anyhow!("Missing score vector for group '{}'", group))?;
        let group_pvals = pvals
            .remove(group)