将矩阵元素映射到EJML中的{0,1}值

时间:2019-02-27 11:44:56

标签: ejml

我想将非负整数矩阵转换为二进制矩阵。例如,给定以下输入矩阵:

2 3
0 1

应为以下输出矩阵:

1 1
0 1

我认为这类似于映射操作,因此在伪代码方面,该操作等效于mapElements(x -> (x > 0) ? 1 : 0)或简称为mapNonZeroes(x -> 1)

一种可能的方法是将矩阵的非零元素展开为三元组,其值设置为0/1,然后从三元组重建矩阵。有更好的方法吗?

1 个答案:

答案 0 :(得分:1)

对我来说,有效的方法是直接访问nz_values存储字段,然后自己映射值。

public void normalizeMatrix(DMatrixSparseCSC m) {
    for (int i = 0; i < m.nz_length; i++) {
        m.nz_values[i] = Math.min(m.nz_values[i], 1.0);
    }
}