Function convert_to_array_f64_csr_selected

src/shared/mod.rs:361–394

(
    csr: &CsrMatrix<T>,
    shape: Shape,
    row_selection: &SelectInfoElem,
    col_selection: &SelectInfoElem,
)

}

fn convert_to_array_f64_csr_selected<T: NumericOps>(
    csr: &CsrMatrix<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 column indices to output column indices
    let col_map: HashMap<usize, usize> = col_indices
        .iter()
        .enumerate()
        .map(|(i, &col)| (col, i))
        .collect();

    for (out_row, &row) in row_indices.iter().enumerate() {
        if row < csr.nrows() {
            let row_start = csr.row_offsets()[row];
            let row_end = csr.row_offsets()[row + 1];
            for (&col, &value) in csr.col_indices()[row_start..row_end]
                .iter()
                .zip(csr.values()[row_start..row_end].iter())
            {
                if let Some(&out_col) = col_map.get(&col) {
                    dense[[out_row, out_col]] = NumCast::from(value)
                        .ok_or_else(|| anyhow!("Failed to convert value to f64"))?;
                }
            }
        }
    }
    Ok(dense)
}

fn convert_to_array_f64_csc_selected<T: NumericOps>(
    csc: &CscMatrix<T>,