Function convert_to_array_f64_csc_selected

src/shared/mod.rs:396–429

(
    csc: &CscMatrix<T>,
    shape: Shape,
    row_selection: &SelectInfoElem,
    col_selection: &SelectInfoElem,
)

}

fn convert_to_array_f64_csc_selected<T: NumericOps>(
    csc: &CscMatrix<T>,
    shape: Shape,
    row_selection: &SelectInfoElem,
    col_selection: &SelectInfoElem,
) -> anyhow::Result<Array2<f64>> {
    let row_indices = select_info_elem_to_indices(row_selection, shape[0])?;
    let col_indices = select_info_elem_to_indices(col_selection, shape[1])?;
    let mut dense = Array2::<f64>::zeros((row_indices.len(), col_indices.len()));

    // Create a mapping from original row indices to output row indices
    let row_map: HashMap<usize, usize> = row_indices
        .iter()
        .enumerate()
        .map(|(i, &row)| (row, i))
        .collect();

    for (out_col, &col) in col_indices.iter().enumerate() {
        if col < csc.ncols() {
            let col_start = csc.col_offsets()[col];
            let col_end = csc.col_offsets()[col + 1];
            for (&row, &value) in csc.row_indices()[col_start..col_end]
                .iter()
                .zip(csc.values()[col_start..col_end].iter())
            {
                if let Some(&out_row) = row_map.get(&row) {
                    dense[[out_row, out_col]] = NumCast::from(value)
                        .ok_or_else(|| anyhow!("Failed to convert value to f64"))?;
                }
            }
        }
    }
    Ok(dense)
}

pub fn convert_to_array_f64_selected(
    data: &ArrayData,