犰狳中 find() 的补充索引

Question

来自 R 的 Armadillo 和 C++ 新手的基本问题。

我有一个向量X，我想将0 下面的条目设置为给定值，将大于0 的条目设置为另一个值。犰狳具有返回indices of elements of X that are non-zero or satisfy a relational condition 的find 函数（不合逻辑！？）所以我可以这样做：

arma::uvec ind0 = find(X < 0);
arma::uvec ind1 = find(X >= 0);

X(ind0).zeros();
X(ind1).fill(1);

这显然不是最好的解决方案。不涉及两次调用find 的更好方法是什么？

Answer 1

您可以使用.transform() 成员函数。需要 C++11 编译器。

mat X(100,100,fill::randu);

X -= 0.5;

X.transform( [](double val) { return (val < 0) ? double(0) : double(1); } );

Answer 2

这就是你需要的！它将使小于 0 的值等于 0，大于 1 的值等于 1。

 X = clamp(X, 0, 1); 

更多信息请访问： http://arma.sourceforge.net/docs.html#clamp