find - 使用带有两个向量的 find() 函数

Question

我使用的是 Julia 编程语言，我知道你可以通过以下方式使用 find 函数:

a = [ 1 2 3 4 3 5 3 6 7 8 9 3 ]
find(a .== 3)

它将返回:3,5,7,12

足够简单。

但是，我的问题是，如果我们想把上面代码中的 3 替换成一个向量怎么办。

例如:

a = [1 2 3 4 5 6 7]
b = [1 9 5 8]

以下语法对我不起作用，但它传达了我的想法。我将如何正确执行以下操作？:

find (a .== b)

如果我们希望它返回 1, 3？

我知道 R 中的匹配函数做得很好，但是我有一个非常大的数据集，而 R 并没有很好地处理它。

Answer 1

另一种方式-使用函数findin/2 :

julia> findin(b,a)
2-element Array{Int64,1}:
 1
 3

julia> findin(a,b)
2-element Array{Int64,1}:
 1
 5