Arguments par défaut

Introduction à l'écriture de fonctions en R

Richie Cotton

Data Evangelist at DataCamp

Problèmes avec toss_coin()

toss_coin <- function(n_flips, p_head) {
  coin_sides <- c("head", "tail")
  weights <- c(p_head, 1 - p_head)
  sample(coin_sides, n_flips, replace = TRUE, prob = weights)
}

Définir la valeur par défaut dans la signature

toss_coin <- function(n_flips, p_head = 0.5) {
  coin_sides <- c("head", "tail")
  weights <- c(p_head, 1 - p_head)
  sample(coin_sides, n_flips, replace = TRUE, prob = weights)
}
Introduction à l'écriture de fonctions en R

Gabarit avec valeurs par défaut

my_fun <- function(data_arg1, data_arg2, detail_arg1 = default1) {
  # Do something
}
Introduction à l'écriture de fonctions en R

Autres types de valeur par défaut

args(median)
function (x, na.rm = FALSE, ...)
library(jsonlite)
args(fromJSON)
function (txt, simplifyVector = TRUE, simplifyDataFrame = simplifyVector, 
    simplifyMatrix = simplifyVector, flatten = FALSE, ...)
Introduction à l'écriture de fonctions en R

Valeurs NULL par défaut

Par convention, cela signifie :

La fonction traite ce paramètre de façon spéciale. Veuillez lire la doc.

args(set.seed)
function (seed, kind = NULL, normal.kind = NULL)
Introduction à l'écriture de fonctions en R

Valeurs catégorielles par défaut

  1. Passer un vecteur de caractères dans la signature.
  2. Appeler match.arg() dans le corps.
args(prop.test)
function (x, n, p = NULL, alternative = c("two.sided", "less", "greater"), 
  conf.level = 0.95, correct = TRUE)

Dans le corps

alternative <- match.arg(alternative)
Introduction à l'écriture de fonctions en R

Découper un vecteur par quantile

cut_by_quantile <- function(x, n, na.rm, labels, interval_type) {
  probs <- seq(0, 1, length.out = n + 1)
  quantiles <- quantile(x, probs, na.rm = na.rm, names = FALSE)
  right <- switch(interval_type, "(lo, hi]" = TRUE, "[lo, hi)" = FALSE)
  cut(x, quantiles, labels = labels, right = right, include.lowest = TRUE)
}
  • x : Vecteur numérique à découper
  • n : Nombre de catégories pour découper x
  • na.rm : Faut-il supprimer les valeurs manquantes ?
  • labels : Étiquettes de caractères pour les catégories
  • interval_type : Intervalles ouverts à gauche ou à droite ?
Introduction à l'écriture de fonctions en R

Poids du coeur de chats

Un diagramme en bandes du poids du coeur de chats. L’étendue va de 3 à 20 grammes.

quantile(cats$Hwt)
    0%    25%    50%    75%   100% 
 6.300  8.950 10.100 12.125 20.500
1 data(cats, package = "MASS")
Introduction à l'écriture de fonctions en R

Découpe par quantile

Le diagramme en bandes précédent est coloré : points rouges à gauche, puis verts, turquoise, puis violets à droite.

cut(x, quantile(x))
Introduction à l'écriture de fonctions en R

Passons à la pratique !

Introduction à l'écriture de fonctions en R

Preparing Video For Download...