julia - 在 Julia 中提取特定的子数组

Question

来自数组

[3,1,7,2;
4,3,2,7;
3,4,1,2]

我想提取与最后一个主菜等于 2 的行对应的子数组。

我是一名尝试开始使用 Julia 的 Matlab 用户。我在文档中查找提示，但未能找到有效的答案。

非常感谢您，

史蒂芬

Answer 1

这对你有用吗？

julia> x = [3 1 7 2
            4 3 2 7
            3 4 1 2]
3x4 Array{Int64,2}:
 3  1  7  2
 4  3  2  7
 3  4  1  2

julia> x[x[:, end] .== 2, :]
2x4 Array{Int64,2}:
 3  1  7  2
 3  4  1  2

让我们分解一下。

x[:, end] 是最后一列。

x[:, end] .== 2 给出的是一个 Vector{Bool}(true 和 的一维数组false)，如果该行以 2 结尾，则为 true，否则为 false。

然后将它们放在一起我们有 x[x[:, end] .== 2, :]，它采用 true 和 的向量false 指定哪些行和 ,: 表示获取每一行中的所有列。