If else di R: if, else if, ifelse(), dan switch()

Pernyataan if

Pernyataan if di R menjalankan sebuah blok kode hanya saat kondisinya TRUE. Kondisinya ditaruh di dalam tanda kurung, kodenya di dalam kurung kurawal, dan else yang opsional menentukan apa yang terjadi kalau tidak:

score <- 72

if (score >= 60) {
  print("You passed")
} else {
  print("Try again")
}

Kondisi score >= 60 dievaluasi menjadi satu TRUE, jadi cabang pertama dijalankan. Operator perbandingan dan logika (==, >=, &&, %in%, dan kawan-kawannya) adalah bahan untuk menyusun kondisi-kondisi ini.

Untuk lebih dari dua kemungkinan hasil, rangkai pemeriksaan dengan else if. R menelusuri rangkaian itu dari kondisi pertama ke bawah dan menjalankan cabang pertama yang kondisinya TRUE, lalu melewati sisanya:

score <- 85

if (score >= 90) {
  grade <- "A"
} else if (score >= 80) {
  grade <- "B"
} else if (score >= 70) {
  grade <- "C"
} else {
  grade <- "F"
}

print(grade)

Urutan itu penting: karena 85 >= 80 diperiksa sebelum 85 >= 70, nilainya "B". Kalau kamu mengurutkan pemeriksaannya sebaliknya, setiap skor di atas 70 akan mendapat "C".

Jebakan kurung kurawal: di mana else harus ditaruh

Inilah error yang menggigit hampir setiap skrip R pertama. Di skrip (atau apa pun yang dijalankan dengan Rscript), else harus muncul di baris yang sama dengan kurung kurawal penutup cabang sebelumnya. Ini gagal:

if (score >= 60) {
  print("You passed")
}
else {              # Error: unexpected 'else'
  print("Try again")
}

Alasannya: R mengurai kode pernyataan demi pernyataan. Saat sampai di } di awal baris, pernyataan if sudah lengkap secara sintaks, dan R langsung mengevaluasinya saat itu juga. else di baris berikutnya datang sebagai awal pernyataan baru, padahal sebuah pernyataan tidak bisa diawali dengan else. Solusinya adalah pola yang dipakai di sepanjang halaman ini:

if (score >= 60) {
  print("You passed")
} else {            # brace and else on the same line
  print("Try again")
}

Di dalam isi fungsi atau sepasang kurung kurawal luar mana pun, aturan ketat ini melonggar (R terus membaca sampai kurung kurawal luarnya ditutup), tapi jangan mengandalkannya: gaya } else { bekerja di mana saja dan itulah yang dianjurkan panduan gaya R.

Kondisinya harus satu TRUE atau FALSE

if() menuntut tepat satu nilai logika. Ada dua hal yang melanggarnya, dan keduanya error:

if (NA) print("hi")
# Error: missing value where TRUE/FALSE needed

if (c(TRUE, FALSE)) print("hi")
# Error: the condition has length > 1

Kasus NA penting dalam praktik karena perbandingan dengan data yang hilang menghasilkan NA: kalau x bernilai NA, maka x > 10 adalah NA, dan if yang diberi kondisi itu menghentikan skripmu. Pola pertahanannya memakai && yang short-circuit:

x <- NA

if (!is.na(x) && x > 10) {
  print("big")
} else {
  print("small or missing")
}

Karena !is.na(x) bernilai FALSE, bagian x > 10 tidak pernah dievaluasi dan tidak ada NA yang sampai ke kondisi. Cara NA menyebar melalui perbandingan dibahas di nilai yang hilang.

Error panjang lebih dari 1 (diberlakukan sejak R 4.2; sebelumnya R diam-diam memakai elemen pertama) biasanya berarti kamu sebenarnya memaksudkan salah satu dari ini: any(v > 5) kalau satu kecocokan sudah cukup, all(v > 5) kalau setiap elemen harus lolos, atau ifelse() kalau kamu menginginkan jawaban per elemen, yang dibahas di bagian berikutnya.

ifelse(): if else yang tervektorisasi

if memutuskan sekali. ifelse(test, yes, no) memutuskan untuk setiap elemen vector dan mengembalikan vector dengan panjang yang sama:

temps <- c(31, 18, 24, 35, 12)

label <- ifelse(temps >= 25, "hot", "mild")
print(label)

Setiap elemen temps diuji secara terpisah: "hot" "mild" "mild" "hot" "mild". Inilah alat untuk menurunkan vector atau kolom data frame baru dari yang sudah ada, tanpa perlu loop.

Tidak seperti if, ifelse() menangani nilai yang hilang dengan anggun: NA di dalam uji menghasilkan NA di hasilnya, bukan error:

temps <- c(31, NA, 12)
print(ifelse(temps >= 25, "hot", "mild"))

Kamu bisa menyarangkan pemanggilan ifelse() untuk pembagian tiga arah, dan pada kedalaman itu kodenya masih mudah dibaca:

temps <- c(31, 18, 24, 35, 12)

band <- ifelse(temps >= 30, "hot",
        ifelse(temps >= 20, "warm", "cool"))
print(band)

Tapi berhentilah di situ. Tiga atau lebih pemanggilan ifelse() bersarang berubah menjadi teka-teki mencocokkan kurung; pada titik itu, pakai cut() (untuk rentang numerik) atau dplyr::case_when(), yang menaruh setiap kondisi di barisnya sendiri.

switch() untuk pilihan bercabang banyak

Saat kamu mencocokkan satu nilai dengan daftar nama yang sudah diketahui (bukan menguji rentang), rangkaian else if yang panjang adalah alat yang canggung. switch() mengatakannya secara langsung:

weekday_type <- function(day) {
  switch(day,
    sat = ,
    sun = "weekend",
    "weekday"
  )
}

print(weekday_type("sun"))
print(weekday_type("tue"))

Tiga perilaku yang perlu diketahui:

• Pencocokan: switch() membandingkan nilai character dengan nama-nama argumen dan mengembalikan hasil yang cocok.
• Fall-through: nama dengan sisi kanan kosong (sat = ,) jatuh ke hasil kasus berikutnya; begitulah "sat" dan "sun" berbagi "weekend" tanpa mengulangnya.
• Default: argumen terakhir tanpa nama ("weekday") dikembalikan saat tidak ada yang cocok. Tanpa default, nilai yang tidak cocok membuat switch() mengembalikan NULL secara tak terlihat: fungsimu diam-diam tidak mengembalikan apa-apa, bug yang membingungkan. Selalu berikan default (atau jadikan stop() yang eksplisit).

switch() bersinar di dalam fungsi yang memilih perilaku berdasarkan argumen mode atau method; untuk apa pun yang berdasarkan ambang numerik, tetap pakai else if atau cut().

Yang kamu bawa pulang

• if (condition) { } else { }: kondisi di dalam tanda kurung, satu cabang dijalankan; rangkai dengan else if, kecocokan pertama yang menang.
• Di skrip, tulis } else { dalam satu baris; else di awal baris adalah error sintaks.
• if() butuh tepat satu TRUE/FALSE: NA adalah error, vector yang lebih panjang adalah error (sejak R 4.2). Lindungi dengan !is.na(x) &&, atau ringkas dengan any()/all().
• ifelse(test, yes, no) adalah versi tervektorisasi; pakai untuk menghitung vector baru, dan jangan menyarangkannya lebih dari dua kali.
• switch() menangani pilihan bercabang banyak berdasarkan nama, dengan fall-through lewat kasus kosong; selalu berikan default.

Berikutnya: melakukan sesuatu lebih dari sekali, yaitu for loop, dan kapan loop adalah alat yang tepat (dan kapan bukan).