Standardargument

Introduktion till att skriva funktioner i R

Richie Cotton

Data Evangelist at DataCamp

Problem med 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)
}

Ange standardvärdet i signaturen

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)
}

En mall med standardvärden

my_fun <- function(data_arg1, data_arg2, detail_arg1 = default1) {
  # Do something
}

Andra typer av standardvärden

args(median)

function (x, na.rm = FALSE, ...)

library(jsonlite)
args(fromJSON)

function (txt, simplifyVector = TRUE, simplifyDataFrame = simplifyVector, 
    simplifyMatrix = simplifyVector, flatten = FALSE, ...)

NULL som standardvärde

Enligt konvention innebär detta

Funktionen hanterar det här argumentet på ett särskilt sätt. Läs dokumentationen.

args(set.seed)

function (seed, kind = NULL, normal.kind = NULL)

Kategoriska standardvärden

Skicka en teckensträng-vektor i signaturen.
Anropa match.arg() i funktionskroppen.

args(prop.test)

function (x, n, p = NULL, alternative = c("two.sided", "less", "greater"), 
  conf.level = 0.95, correct = TRUE)

I funktionskroppen

alternative <- match.arg(alternative)

Dela upp en vektor efter kvantil

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: En numerisk vektor att dela upp
n: Antal kategorier att dela upp x i
na.rm: Ska saknade värden tas bort?
labels: Textetiketter för kategorierna
interval_type: Ska intervallen vara öppna till vänster eller höger?

Katters hjärtvikter

Ett punktdiagram över katters hjärtvikter. Spannet är 3 till 20 gram.

quantile(cats$Hwt)

    0%    25%    50%    75%   100% 
 6.300  8.950 10.100 12.125 20.500

¹ data(cats, package = "MASS")

Dela upp efter kvantil

Punktdiagrammet från föregående slide har fått färg: röda punkter till vänster, sedan gröna, sedan turkosa och lila till höger.

cut(x, quantile(x))

Nu kör vi en övning!

Introduktion till att skriva funktioner i R