[Julia]dataframeに一定期間の平均値を出した列を追加する際に条件を追加する方法

Question

juliaのデータフレームにC列平均値を記載した新規列を追加する場合下記のコードで可能ですが、B列の値を参照する方法はありますでしょうか？
具体的には列Bの値が2の場合は列Bの値が1の値は参照しない方法をお教えいただけると幸いです。

data = DataFrame(
    A = 1:10,
    B = [1,1,1,1,1,2,2,2,2,2],
    C = rand(10),
)

n = 3

data[!, "sma(n=$n)"] = sma(data.C, n=n)

下記に現在作成されるデータフレームと意図しているデータフレームの画像を添付します。

Answer 1

transform + groupby を使います。

n = 3
data = transform(groupby(data, :B), :C => (x -> sma(x[:], n=n)) => "sma(n=$n)")
print(data)

#
10×4 DataFrame
 Row │ A      B      C         sma(n=3)
     │ Int64  Int64  Float64   Float64
─────┼────────────────────────────────────
   1 │     1      1  0.622326  NaN
   2 │     2      1  0.184663  NaN
   3 │     3      1  0.772625    0.526538
   4 │     4      1  0.16359     0.373626
   5 │     5      1  0.455608    0.463941
   6 │     6      2  0.396216  NaN
   7 │     7      2  0.983158  NaN
   8 │     8      2  0.631562    0.670312
   9 │     9      2  0.471443    0.695388
  10 │    10      2  0.891725    0.66491

Comments

[akirayoshimura]

回答ありがとうございます。
Juliaってこんなことも出来るんですね！
勉強になりました、ありがとうございます。