创建新矩阵，所有非零元素都设置为1(julia)
我想要一种将 julia 中的矩阵处理为新矩阵的有效（矢量化）方法，其中所有非零元素都是新矩阵中的元素。
例如，我想要这个矩阵
0 3 6 8 0
7 0 2 0 1
0 4 9 1 0
成为
0 1 1 1 0
1 0 1 0 1
0 1 1 1 0
回答
最简单的方法是将此矩阵转换为如下矩阵BitMatrix：
julia> x = [0 3 6 8 0
7 0 2 0 1
0 4 9 1 0]
3×5 Matrix{Int64}:
0 3 6 8 0
7 0 2 0 1
0 4 9 1 0
julia> x .!= 0
3×5 BitMatrix:
0 1 1 1 0
1 0 1 0 1
0 1 1 1 0
如果您希望矩阵包含 Int值，则可以执行以下操作：
julia> Int.(x .!= 0)
3×5 Matrix{Int64}:
0 1 1 1 0
1 0 1 0 1
0 1 1 1 0
最后如果你 0值具有混合类型并且您希望“按原样”保留它们，那么您可以执行以下操作：
julia> x = Real[0 3 6 8 UInt8(0)
7 0 2 0 1
0.0 4 9 1 0]
3×5 Matrix{Real}:
0 3 6 8 0x00
7 0 2 0 1
0.0 4 9 1 0
julia> @. ifelse(iszero(x), x, 1)
3×5 Matrix{Real}:
0 1 1 1 0x00
1 0 1 0 1
0.0 1 1 1 0
除了广播之外，您还可以使用理解，这也应该很快。例如。
julia> [v == 0 ? v : 1 for v in x]
3×5 Matrix{Real}:
0 1 1 1 0x00
1 0 1 0 1
0.0 1 1 1 0
或例如
julia> [ifelse(v == 0, 0, 1) for v in x]
3×5 Matrix{Int64}:
0 1 1 1 0
1 0 1 0 1
0 1 1 1 0