R 時系列データの、ある1時点の値が入った変数から過去3年の合計値が入った変数を作りたいです

Question

前提・実現したいこと

使用したいデータは以下のような時系列データです。
name year stockprice
A社 1999 200
A社 2000 199
A社 2001 201
A社 2002 205
B社 1999 ...
B社 2000 ...
B社 2001 ...

以上のデータのstockpriceという列のデータから、過去3年の合計値が入ったtotal_stockpriceという変数を作成したいです。
例えばA社2001年の場合、total_stockprice=200+199+201=600、
A社2002年の場合、total_stockprice=199+201＋205=605となります

どのような関数を使えばよいか調べてもわからなかったため、教えていただけると嬉しいです。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

データフレームに対して、特定の項目に書かれた種別ごとに何らかの処理をしたい場合、tapply関数が使えます。
yearを1999以上2001以下に限定したうえで、会社ごとにpriceの合計を得たいとすると、ざっとこんな感じです。

df <- data.frame(
  "name" = c(rep("A社", 4), rep("B社", 4)),
  "year"  = rep(1999:2002, 2),
  "stockprice" = c(200, 199, 201, 205, 300, 299, 301, 305)
)
df$name.f <- as.factor(df$name)

df1 <- subset(df, 1999<=year & year<=2001)
tapply(df1$stockprice, df1$name.f, sum)

実際には上記のコードに基づいてもう少し汎用性を持たせたユーザー関数を仕立てるのがいいでしょう。

Comments

[Mayu___]

ご回答いただきありがとうございます！参考にさせていただきます

Answer 2

R のデフォルト関数のみを使ってみました(dplyr パッケージなどを利用すれば、もっと判りやすい書き方ができるかと思います)。

df <- data.frame(
  "name" = c(rep("A社", 4), rep("B社", 4)),
  "year"  = rep(1999:2002, 2),
  "stockprice" = c(200, 199, 201, 205, 300, 299, 301, 305)
)

do.call(rbind, c(
  lapply(split(df, df$name),
    function(x) {
      data.frame(x,
        total_stockprice = {
          i = findInterval(x$year-3, x$year)
          w = which(as.logical(i))
          cs = cumsum(x$stockprice)  
          i[w] = cs[i[w]]
          cs - i
      })}), make.row.names=F))

df

#
df
  name year stockprice total_stockprice
1  A社 1999        200              200
2  A社 2000        199              399
3  A社 2001        201              600
4  A社 2002        205              605
5  B社 1999        300              300
6  B社 2000        299              599
7  B社 2001        301              900
8  B社 2002        305              905

Comments

[Mayu___]

ご回答いただきありがとうございます！
実際のデータ（2002年以降含む）で応用できそうなので、真似させていただいてます。

追加で質問させていただきたいのですが、
スクリプト1行目でdf <- data.frame(...と、dfの中身を定義していますが、既存のデータを使う場合、9行目data.frame(x,...の「data.frame」の部分はどのようにしたらよいでしょうか？
試しに実際の時系列データ名"sorted_ALLYEAR"を該当箇所に入れてみたのですが、「sorted_ALLYEAR(x, total_stockprice = { でエラー: 　　 関数 "sorted_ALLYEAR" を見つけることができませんでした 」というエラーメッセージが表示されている状態です。

上手く説明できていない部分があったら申し訳ございません。ご回答いただけると大変助かります。よろしくお願いいたします。

[melian]

sorted_ALLYEAR を入れるのそこではなく、lapply の箇所です。

do.call(rbind, c(
lapply(split(sorted_ALLYEAR, sorted_ALLYEAR$name),
function(x) { ...

function(x) { ... 以降は変更しません。

[Mayu___]

ご返信ありがとうございます！
お陰様で上手くいきました^ ^
大変助かりました