IF, ELSE, ELSE IF uttalande i R

if else-satsen i R

An om annat En sats kör ett kodblock när ett villkor är SANT och ett annat när det är FALSKT. I R är syntaxen:

if (condition) {
    Expr1 
} else {
    Expr2
}

Kontrollera om en variabel med namnet kvantitet är över 20. Om den är det skriver koden ut "Du sålde mycket!", annars skriver den ut "Inte tillräckligt för idag".

# Create vector quantity
quantity <-  25
# Set the if-else statement
if (quantity > 20) {
    print('You sold a lot!')
} else {
    print('Not enough for today')  
}

Produktion:

## [1] "You sold a lot!"

AnmärkningarIndragning ändrar inte hur R kör koden, men konsekvent indentering är det som gör en lång kedja av villkor läsbar. Observera också att den avslutande klammerparentesen och nyckelordet else måste stå på samma rad, annars behandlar R if-blocket som komplett och ger ett fel.

Else if-satsen i R

Kedja ytterligare tester med annars omR har inget elif-nyckelord, det vill säga Python; i R skriver man de två orden separat, och man kan lägga till så många grenar som man behöver. Syntaxen är:

if (condition1) { 
    expr1
    } else if (condition2) {
    expr2
    } else if  (condition3) {
    expr3
    } else {
    expr4
}

Anta att du vill ha tre resultat: under 20 utskrifter "Inte tillräckligt för idag", mellan 20 och 30 utskrifter "Genomsnittlig dag", och över 30 utskrifter "Vilken fantastisk dag!".

Du kan försöka ändra mängden.

# Create the variable quantity
quantity <-  10
# Create multiple condition statement
if (quantity < 20) {
      print('Not enough for today')
} else if (quantity >= 20 & quantity <= 30) {
     print('Average day')
} else {
      print('What a great day!')
}

Produktion:

## [1] "Not enough for today"

Exempel 2:

Momssatserna skiljer sig åt beroende på produktkategori. Tänk dig tre kategorier, var och en med sin egen sats:

En annan om-kedja tillämpar rätt pris på vad kunden än köpte:

category <- 'A'
price <- 10
if (category =='A'){
  cat('A vat rate of 8% is applied.','The total price is',price *1.08)  
} else if (category =='B'){
    cat('A vat rate of 10% is applied.','The total price is',price *1.10)  
} else {
    cat('A vat rate of 20% is applied.','The total price is',price *1.20)  
}

Produktion:

## A vat rate of 8% is applied. The total price is 10.8

Den vektoriserade funktionen ifelse() i R

if-satsen hanterar exakt ett värde. Ge den en vektor och R 4.2 eller senare genererar ett fel, eftersom villkoret måste vara ett enda SANT eller FALSKT. ifelse() löser detta genom att testa varje element samtidigt och returnera en vektor av samma längd.

quantities <- c(10, 25, 40, 18)

# if cannot do this
result <- ifelse(quantities > 20, 'You sold a lot!', 'Not enough for today')
result

Produktion:

## [1] "Not enough for today" "You sold a lot!"      "You sold a lot!"      
## [4] "Not enough for today"

De tre argumenten är testet, värdet som returneras där testet är SANT och värdet som returneras där det är FALSKT. Alla tre återvinns till samma längd.

Att nästla ifelse() reproducerar en else if-kedja över en hel vektor:

ifelse(quantities < 20, 'Not enough for today',
    ifelse(quantities <= 30, 'Average day', 'What a great day!'))

Två begränsningar är värda att känna till. ifelse() tar bort attribut som faktornivåer och dateklass från resultatet, och den utvärderar båda grenarna för varje element, så det är slöseri när en gren är dyr. Alternativet till dplyr case_when() undviker kapslingen helt och behåller typerna intakta:

library(dplyr)
case_when(
    quantities < 20  ~ 'Not enough for today',
    quantities <= 30 ~ 'Average day',
    TRUE             ~ 'What a great day!'
)

switch()-satsen i R

När varje gren testar samma variabel för likhet, blir en long else if-kedja svår att läsa. switch() ersätter den med en enda sökning.

category <- 'A'
price <- 10

vat <- switch(category,
    'A' = 1.08,
    'B' = 1.10,
    'C' = 1.20,
    1.20)          # the unnamed last value is the default

cat('The total price is', price * vat)

Tre regler styr dess beteende:

• Med ett teckenargument matchar switch() namnet exakt. Det finns ingen partiell matchning och ingen mönstermatchning.
• Ett namnlöst slutelement fungerar som standard. Utan ett sådant returnerar ett omatchat värde NULL osynligt, vilket är en vanlig källa till tysta buggar.
• Med ett heltalsargument väljer switch() istället efter position: switch(2, "a", "b", "c") returnerar "b".

Använd switch() för exakta matchningar mot en fast uppsättning värden, och behåll else if för intervall och sammansatta villkor, vilka switch() inte kan uttrycka. För att tillämpa ett villkor upprepade gånger över en sekvens, kombinera det med en för slinga eller ett medan slingan.

logisk Operatorer i R-förhållanden

Operatorerna som bygger ett villkor beter sig olika från varandra, och skillnaden spelar roll.

Varför kortslutning är viktigt. Dubbelformen upphör så snart svaret är avgjort, så höger sida utvärderas aldrig när vänster sida redan bestämmer resultatet. Det gör det säkert att skriva en skyddsklausul:

# Safe: length() is checked before x[1] is touched
if (length(x) > 0 && x[1] > 10) print('ok')

Jämförelse mot flera värden. Använd %in% istället för att kedja ihop likhetstester, vilket är både kortare och svårare att felskriva:

if (category %in% c('A', 'B')) {
    print('Reduced VAT rate applies')
}

En sista varning: ett enda likhetstecken tilldelar ett värde, det jämför inte. Skriv alltid == inuti ett villkor.

Villkorliga satser i R: Snabbreferens

Varje konstruktion som används i den här handledningen listas nedan:

if (cond) { a } else { b }

if (c1) { a } else if (c2) { b } else { c }

ifelse(test, yes, no)

case_when(c1 ~ a, c2 ~ b, TRUE ~ c)

switch(x, "A" = 1, "B" = 2, 0)

if (x %in% c("A", "B")) { }

Vanliga frågor