R enum alike avec match.arg: Comment choisir un élément non-premier comme valeur par défaut?

Question

Comment puis-je déclarer une valeur autre que la première de plusieurs valeurs d'arguments autorisées comme valeur par défaut (si aucune valeur n'a été transmise)?

library(stats) 

center <- function(x, type = c("mean", "median", "trimmed")) { 
    type <- match.arg(type) 
    switch(type, 
     mean = mean(x), 
     median = median(x), 
     trimmed = mean(x, trim = .1)) 
} 

values <- c(100, 120, 200) 

center(values, "median") 
# [1] 120 

center(values, "mean") 
# [1] 140 

# uses the first value "mean" as the default! 
center(values) 
# [1] 140 

Je voudrais voir la valeur par défaut dans la signature de la fonction, si possible, afin de rendre la valeur par défaut explicite (visible), e. g. (! Pseudo-code non-travail) pour faire le deuxième élément de la valeur par défaut:

center <- function(x, type = c("mean", "median", "trimmed")[2]) {... 

D'autres propositions sont également les bienvenus!

PS: Je pourrais mettre la valeur par défaut toujours dans la première position, mais s'il y a un ordre logique dans la liste, je préférerais le garder (par exemple payment.frequency = c("monthly", "quarterly", "semi-annually", "annually")

Answer 1

Vous pouvez modifier match.arg prendre la deuxième arg. valeur au lieu de la première. Il est juste une question de trouver et de remplacer cette ligne dans le corps de la fonction.

matchArg <- base::match.arg 
body(matchArg)[[c(4, 3, 2, 3)]] <- quote(return(arg[2L])) 

maintenant, nous remplacer match.arg avec notre nouvelle fonction matchArg intérieur center, et nous allons maintenant obtenir la deuxième type valeur (la médiane) par défaut.

center <- function(x, type = c("mean", "median", "trimmed")) { 
    type <- matchArg(type) 
    switch(type, 
     mean = mean(x), 
     median = median(x), 
     trimmed = mean(x, trim = .1)) 
} 

values <- c(100, 120, 200) 

center(values) 
# [1] 120 
center(values, "mean") 
# [1] 140 

Bien sûr, cela ne répond pas à vos exigences de visibilité. Mais c'est une idée.

Answer 2

Merci à la réponse acceptée de @RichScriven qui a montré comment modifier match.arg j'ai trouvé un moyen de annoter aussi la valeur par défaut dans la signature de la fonction (qui était un souhait facultatif dans ma question).

L'astuce consiste à "annoter" la valeur par défaut en utilisant un nom pour l'élément vectoriel:

matchArg <- base::match.arg 
body(matchArg)[[c(4, 3, 2, 3)]] <- quote(
    return(
    if ("default" %in% names(arg)) { 
     arg[["default"]] } 
    else { 
     arg[[1L]] 
    }) 
) 

center <- function(x, type = c("mean", default = "median", "trimmed")) { 
    type <- matchArg(type) 
    switch(type, 
     mean = mean(x), 
     median = median(x), 
     trimmed = mean(x, trim = .1)) 
} 

Maintenant, la valeur par défaut est le second (annoté):

> values <- c(100, 120, 200) 
> center(values, "median") 
[1] 120 
> center(values, "mean") 
[1] 140 
> center(values) 
[1] 120 

PS: Cette solution vaudrait un patch pour la base R si suffisamment d'utilisateurs pensent que cette fonctionnalité est nécessaire ...