カテゴリ別の支出

CoddyのRジャーニー「ロジックとフロー」セクションの一部。レッスン 59/64。

チャレンジ

簡単

CATEGORYコマンドを追加してください。これまでの手順にあるすべてのコマンドが、以前とまったく同じように動作し続けるようにしてください。

CATEGORY：カテゴリごとに1行ずつ、アルファベット順で、food: 20.50のように、各カテゴリの合計を小数点以下2桁で返します。何も記録されていない場合は、no expensesと返します。

1つのCATEGORYコマンドから複数の返答が生成され、それぞれが別々の行に出力されます。

提供されたコードは、入力された各行を1つのコマンドとしてtracker()に渡し、返された各応答をそれぞれ別の行に出力します。日付はYYYY-MM-DD形式で記述され、カテゴリは小文字の単語であり、入力にはここまでで説明したコマンドのみが含まれます。

自分で試してみよう

tracker <- function(commands) {
  expenses <- data.frame(date = character(0), category = character(0), amount = numeric(0))
  replies <- c()
  for (cmd in commands) {
    parts <- strsplit(cmd, " ")[[1]]
    if (parts[1] == "ADD") {
      amount <- suppressWarnings(as.numeric(parts[4]))
      if (length(parts) != 4 || is.na(amount) || amount <= 0) {
        replies <- c(replies, paste("invalid:", cmd))
      } else {
        expenses <- rbind(expenses, data.frame(date = parts[2], category = parts[3], amount = amount))
        replies <- c(replies, sprintf("added %s %.2f", parts[3], amount))
      }
    } else if (parts[1] == "TOTAL") {
      replies <- c(replies, sprintf("total: %.2f, count: %d", sum(expenses$amount), nrow(expenses)))
    }
  }
  replies
}

# 提供された入出力コード: そのままにしておく
input <- suppressWarnings(readLines(file("stdin")))
cat(tracker(input), sep = "\n")