如何在Rust ndarray中按列序迭代二维数组？

ndarray crate 完全支持按列优先迭代，不需要自行实现，核心是用轴迭代器来处理：

• 使用 axis_iter(Axis(n))（不可变）或 axis_iter_mut(Axis(n))（可变）方法，指定迭代的轴即可。对于你形状为 (2, N) 的双声道数组，轴1对应列维度，遍历轴1就能按列优先的顺序访问每一组双声道样本。

示例代码 #

假设你的数组是行0存通道1样本、行1存通道2样本：

use ndarray::{Array2, Axis};
use std::fs::File;
use std::io::Write;

fn main() {
    // 构造测试数组：行0=[1,3,5,7]（通道1），行1=[2,4,6,8]（通道2）
    let audio = Array2::<i16>::from_shape_vec((2, 4), vec![1, 2, 3, 4, 5, 6, 7, 8]).unwrap();

    // 生成交错的样本序列（通道1→通道2，依次遍历每一列）
    let interleaved: Vec<i16> = audio.axis_iter(Axis(1))
        .flat_map(|col| col.iter().copied())
        .collect();
    // 结果：[1,2,3,4,5,6,7,8]，完全符合音频文件的交错格式要求

    // 直接写入文件（以小端字节序为例）
    let mut file = File::create("interleaved_audio.raw").unwrap();
    for col in audio.axis_iter(Axis(1)) {
        for &sample in col.iter() {
            file.write_all(&sample.to_le_bytes()).unwrap();
        }
    }
}

关键说明 #

• ndarray 默认是行优先内存布局，但 axis_iter 不受内存布局限制，会严格按指定轴的逻辑顺序迭代，所以即使是行优先存储的数组，也能通过 axis_iter(Axis(1)) 正确按列遍历。
• 如果需要修改样本数据，把 axis_iter 换成 axis_iter_mut 即可，迭代项是列的可变视图，支持修改其中的元素。

内容的提问来源于stack exchange，提问作者Jack Geraghty