Function get_group_indices

src/memory/processing/diffexp/mod.rs:890–961

Get row indices for cells belonging to a specific group. Searches through the groupby column and returns the indices of all cells that match the specified group label. Handles different column data types including strings, integers, and categorical data. # Arguments `adata` - The AnnData object `groupby` - Column name in `adata.obs` containing group labels `group` - Group label to search for #

(adata: &IMAnnData, groupby: &str, group: &str)

/// - No cells found for the specified group
/// - Data type conversion fails
fn get_group_indices(adata: &IMAnnData, groupby: &str, group: &str) -> anyhow::Result<Vec<usize>> {
    let group_col = adata.obs().get_column_from_df(groupby)?;

    let indices = match group_col.dtype() {
        DataType::String => {
            let string_col = group_col.str()?;
            let mut indices = Vec::new();

            for i in 0..string_col.len() {
                if let Some(value) = string_col.get(i) {
                    if value == group {
                        indices.push(i);
                    }
                }
            }
            indices
        }
        DataType::Int8
        | DataType::Int16
        | DataType::Int32
        | DataType::Int64
        | DataType::UInt8
        | DataType::UInt16
        | DataType::UInt32
        | DataType::UInt64 => {
            let target = group.parse::<i64>().map_err(|_| {
                anyhow::anyhow!(
                    "Failed to parse group '{}' as integer for numeric column",
                    group
                )
            })?;

            let mut indices = Vec::new();
            let int_col = group_col.i64()?;
            for i in 0..int_col.len() {
                if let Some(value) = int_col.get(i) {
                    if value == target {
                        indices.push(i);
                    }
                }
            }
            indices
        }
        DataType::Categorical(_, _) => {
            let string_col = group_col.cast(&DataType::String)?;
            let string_col = string_col.str()?;
            let mut indices = Vec::new();

            for i in 0..string_col.len() {
                if let Some(value) = string_col.get(i) {
                    if value == group {
                        indices.push(i);
                    }
                }
            }

            indices
        }