Julia でデータフレーム のあるカラムとリスト内の要素とが一致する部分を抽出したい

Question

Julia でデータフレームから、あるカラム内の要素のと、あるリスト内の要素が一致する部分を抽出したいです。
Rでは以下のような処理です。

> df = data.frame(x=c("a","b","c"),y=c("d","e","f"))
> ext = c("e","f")
> subset(df , y %in% ext)
  x y
2 b e
3 c f

ご教授お願いします。

Answer 1

DataFrames.jl を使った場合です。

using DataFrames
df = DataFrame(x = 'a':'c', y = 'd':'f')
ext = 'e':'f'
df[df.y .∈ (ext,), :]

ext をタプルにしているのは .∈ のブロードキャスティングに対してスカラーとして扱うためです。最後の行は

df[df.y .∈ Ref(ext), :]

でも動きます。(参考：https://docs.julialang.org/en/stable/manual/arrays/#Broadcasting-1)

Comments

[shotaroh]

ありがとうございます。
` ∈`を使うんですね。
キーバインドの設定をしなくてはいけませんね。

[machakann]

'∈' は REPL や jupyter notebook 上なら \in<Tab> と入力すると出ると思います。

[Paalon]

`in`も`∈`と等価です。

```julia
in == ∈
```

[machakann]

実体は同じ関数ですね。in を使う場合は以下のようになります。

using DataFrames
df = DataFrame(x = 'a':'c', y = 'd':'f')
ext = 'e':'f'
df[in.(df.y, (ext,)), :]

[machakann]

in も演算子としてつかえます、これは R の見た目に近いですね。
ただし、ブロードキャストする場合は関数表記になります。

OK
'a' in ['a', 'b', 'c'] # true

NOT OK
['a', 'b'] .in ['a', 'b', 'c'] # error

OK
in.(['a', 'd'], (['a', 'b', 'c'],)) # [true, false]
['a', 'd'] .∈ (['a', 'b', 'c'],) # [true, false]

[shotaroh]

なるほど、in も使うことができるのですね。
お二方とも、ありがとうございます！