import numpy as np
a = np.full((5, 4), 3.)
a[::2, :] = 2.
a[:, 1] = np.sin(a[:, 1]) + 1.
a[a < 1.5] = 4.
odd_sum = a[:, 1::2].sum()
b = np.exp(np.arange(4))
c = a + b
d = c.T.dot(a)
|
#[macro_use]
extern crate ndarray;
use ndarray::prelude::*;
let mut a = Array2::<f64>::from_elem((5, 4), 3.);
a.slice_mut(s![..;2, ..]).fill(2.);
a.column_mut(1).mapv_inplace(|x| x.sin() + 1.);
a.mapv_inplace(|x| if x < 1.5 { 4. } else { x });
let odd_sum = a.slice(s![.., 1..;2]).sum();
let b = Array::from_shape_fn(4, |i| (i as f64).exp());
let c = a + &b;
let d = c.t().dot(&c);
|