集合演算

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

3つの関数は、2つのベクトルを集合として組み合わせます。それぞれは重複しない値を返します：

union(a, b) はどちらか一方のベクトルに含まれる値、intersect(a, b) は両方に含まれる値、そして setdiff(a, b) は b に含まれない a の値を持ちます。

mon <- c("ana", "ben", "cy")
tue <- c("ben", "dee", "cy", "ben")
print(union(mon, tue))
print(intersect(mon, tue))
print(setdiff(mon, tue))

出力：

[1] "ana" "ben" "cy"  "dee"
[1] "ben" "cy" 
[1] "ana"

setdiff() は引数の順序に依存します。引数を入れ替えると、2番目のベクトルにのみ含まれる値を取得できます：

mon <- c("ana", "ben", "cy")
tue <- c("ben", "dee", "cy")
print(setdiff(tue, mon))

出力：

[1] "dee"

何も残っていない場合、結果は空のベクトルになります。テキストの場合は character(0) です。

print(intersect(c("a", "b"), c("x", "y")))

出力：

character(0)

結果は最初のベクトルの順序を維持します。出力に予測可能な順序が必要な場合はソートし、結合する前に結果が空でないか確認します。

a <- c("pear", "fig", "kiwi")
b <- c("kiwi", "pear")
both <- sort(intersect(a, b))
print(paste(both, collapse = ", "))
print(length(setdiff(b, a)) == 0)

出力:

[1] "kiwi, pear"
[1] TRUE

チャレンジ

簡単

compare_courses(a, b)を完成させてください。aとbは、2つのコースの学生です。次の3行を返してください：

Both: に続けて、両方のコースに在籍する学生
Only first: に続けて、aのみに在籍する学生
Total: に続けて、両方のコース全体で異なる学生の人数

各リストの名前を並べ替え、, で結合してください。空のリストはnoneと記述します。

提供されたコードは、2つのコースをカンマ区切りの行として読み込み、返された各行を出力します。

自分で試してみよう

compare_courses <- function(a, b) {
  # ここにコードを書いてください
  c("Both: ", "Only first: ", "Total: 0")
}

# 提供された入出力コード：そのままにしておいてください
input <- suppressWarnings(readLines(file("stdin")))
cat(compare_courses(strsplit(input[1], ",")[[1]], strsplit(input[2], ",")[[1]]), sep = "\n")