C++ Armadillo库中如何选中矩阵单整列并进行批量操作？

Armadillo完全支持你说的列全选操作，语法和MATLAB高度相似，有两种常用实现方式：

• 第一种是调用内置的.col()成员方法，注意Armadillo的索引默认从0开始，和MATLAB的1起始索引有区别。

你给出的MATLAB示例对应Armadillo代码如下：

#include <armadillo>
using namespace arma;

int main()
{
    // 初始化5*5全1矩阵
    mat A = ones<mat>(5,5);
    // 选中第0列（对应MATLAB中第1列）所有元素乘5
    A.col(0) *= 5;
    // 打印矩阵
    A.print("A = ");
    return 0;
}

运行上述代码得到的输出和你贴的MATLAB运行结果完全一致。

• 第二种是更贴近MATLAB冒号语法的写法，用span::all指代选中某一维度的全部元素：

// 效果和.col()写法完全相同，更适配MATLAB用户的使用习惯
A(span::all, 0) *= 5;

两种写法性能没有区别，.col()写法更简洁，span写法可以灵活扩展到选中部分行的场景，比如A(span(1,3), 0)就可以选中第1到第3行的第0列元素。如果需要选取多列，也可以调用.cols(start_col, end_col)方法实现。

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