arrays - Julia，使用 findall 重置 3d 数组中的第三行

Question

我试图用来findall获取一个一维数组的哪些元素大于第二个一维数组的元素的索引，然后使用这些索引将第三个一维数组的相应值设置为 0。MWE：

# create 3d array
a, b = [3;2;2], [4;3;2];
c = transpose(cat(a,b, dims = 2));
d, e = [1;2;3], [2;3;4];
f = transpose(cat(d,e, dims = 2));
g = cat(c, f, dims = 3);

g
2×3×2 Array{Int64,3}:
[:, :, 1] =
 3  2  2
 4  3  2

[:, :, 2] =
 1  2  3
 2  3  4

findall.(g[end,:,1] >= g[end-1,:,1]) 

并使用索引来重置 g[end,:,2] 的元素，这样我最终得到

g
2×3×2 Array{Int64,3}:
[:, :, 1] =
 3  2  2
 4  3  2

[:, :, 2] =
 1  2  3
 0  0  4

谢谢。Ĵ

Answer 1

下面的代码给出了您要求的答案。你只是.在错误的地方。您想>逐个元素地比较操作，然后应用于findall整个结果数组（而不是逐个元素）。

julia> g[end, findall(g[end,:,1] .> g[end-1,:,1]), 2] .= 0
2-element view(::Array{Int64,3}, 2, [1, 2], 2) with eltype Int64:
 0
 0

julia> g
2×3×2 Array{Int64,3}:
[:, :, 1] =
 3  2  2
 4  3  2

[:, :, 2] =
 1  2  3
 0  0  4

但是，我不会尝试将您的所有数据编译成这样的一个大数组。使用三个单独的一维数组变量比在一个变量中使用三个维度更容易。再次使用上面的变量：

julia> e[b .> a] .= 0
2-element view(::Array{Int64,1}, [1, 2]) with eltype Int64:
 0
 0

julia> e
3-element Array{Int64,1}:
 0
 0
 4