lair 0.8.0

Linear algebra in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use ndarray::{ArrayBase, Data, Ix2};

use crate::Scalar;

/// Finds the last non-zero column in a matrix.
pub fn ilalc<A, S>(x: &ArrayBase<S, Ix2>) -> Option<usize>
where
    A: Scalar,
    S: Data<Elem = A>,
{
    if x.nrows() == 0 || x.ncols() == 0 {
        return None;
    }

    let last_col = x.ncols() - 1;
    if x[(x.nrows() - 1, last_col)] != A::zero() {
        return Some(last_col);
    }

    (0..x.ncols())
        .rev()
        .find(|&col| x.column(col).iter().any(|&elem| elem != A::zero()))
}

/// Finds the last non-zero row in a matrix.
pub fn ilalr<A, S>(x: &ArrayBase<S, Ix2>) -> Option<usize>
where
    A: Scalar,
    S: Data<Elem = A>,
{
    if x.nrows() == 0 || x.ncols() == 0 {
        return None;
    }

    let last_row = x.nrows() - 1;
    if x[(last_row, x.ncols() - 1)] != A::zero() {
        return Some(last_row);
    }

    (0..x.nrows())
        .rev()
        .find(|&row| x.row(row).iter().any(|&elem| elem != A::zero()))
}

#[cfg(test)]
mod tests {
    use ndarray::{arr2, Array2};

    #[test]
    fn ilalc_empty() {
        let x = Array2::<f32>::eye(0);
        assert!(super::ilalc(&x).is_none());
    }

    #[test]
    fn ilalc_zero() {
        let x = Array2::<f32>::zeros((2, 3));
        assert!(super::ilalc(&x).is_none());
    }

    #[test]
    fn ilalc() {
        let x = arr2(&[[0., 1., 0.], [0., 0., 0.]]);
        assert_eq!(super::ilalc(&x), Some(1));
    }
}