Spaces:
Build error
Build error
File size: 1,369 Bytes
84d2a97 |
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 67 |
/// Convert a shape [N, M] to [M, N].
///
/// # Example
///
/// [
/// [1, 2, 3],
/// [4, 5, 6],
/// ]
///
/// to
///
/// [
/// [1, 4],
/// [2, 5],
/// [3, 6],
/// ]
///
/// # Panics
///
/// May panic if the input is not a rectangle.
pub fn transpose<T>(v: Vec<Vec<T>>) -> Vec<Vec<T>> {
transposed_iter(v).collect()
}
/// Convert a shape [N, M] to an iterator which collects into [M, N].
///
/// # Example
///```text
/// [
/// [1, 2, 3],
/// [4, 5, 6],
/// ]
///
/// to
///
/// [1, 4] -> [2, 5] -> [3, 6]
///```
/// # Panics
///
/// May panic if the input is not a rectangle.
pub fn transposed_iter<T>(rectangle: Vec<Vec<T>>) -> impl Iterator<Item = Vec<T>> {
assert!(!rectangle.is_empty());
let len = rectangle.first().map(Vec::len).unwrap_or(0);
let mut iters: Vec<_> = rectangle.into_iter().map(|n| n.into_iter()).collect();
(0..len).map(move |_| {
iters
.iter_mut()
.map(|n| n.next().expect("Input is rectangular"))
.collect()
})
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_transpose() {
let v = vec![vec![1, 2, 3], vec![4, 5, 6]];
let res = transpose(v);
let expected = [vec![1, 4], vec![2, 5], vec![3, 6]];
for (i, column) in res.iter().enumerate() {
assert_eq!(column, &expected[i]);
}
}
}
|