Regroupement de fonctions

Version imprimable

Salut à tous!
J'ai 2 fonctions à regrouper en une seule mais je n'arrive pas à trouver comment faire...
Voici mes 2 fonctions :

Code:

1
2
3
4
5
6
imc=rep(0,100)
imc = sapply(1:100, function(i){
          poids=rnorm(10,60,5)
          taille=rnorm(10,170,5)
          imc=poids/((taille/100)*(taille/100))
          })

Code:

1
2
3
4
5
6
fdr=rep(0,100)
fdr = sapply(1:100, function(i){
          sucre=rbinom(10,1,0.2)
          sel=rbinom(10,1,0.3)
          fdr=sucre+sel
          })

:merci: par avance!

25/08/2011, 11h10
arm3366

Salut,
qu'est ce que tu appelles fonction ici?
25/08/2011, 11h15
NemoParis

Salut arm,

j'appelle fonction mes deux programmes qui me retournent les résultats "imc" et "fdr"
C'est vrai que le terme fonction n'est paut-être pas approprié!

Bonjour,

Le problème c'est que tu ne peux pas inclure les deux dans le même sapply, à moins de faire un post-traitement sur une liste.

Tu peux néanmoins regrouper les deux sapply dans une même fonction qui te retourne une liste contenant les deux objets :

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
myfun.f <- function(n)
{
    ## IMC :
    imc <- sapply(1:n,
                  function(i)
              {
                  poids <- rnorm(10, 60, 5)
                  taille <- rnorm(10, 170, 5)
 
                  return(poids / ((taille/100) * (taille/100)))
              })
 
    ## FDR :
    fdr <- sapply(1:n,
                  function(i)
              {
                  sucre <- rbinom(10,1,0.2)
                  sel <- rbinom(10,1,0.3)
 
                  return(sucre + sel)
              })
 
    ## Retour des résultats :
    return(list(imc=imc, fdr=fdr))
}

À utiliser comme suit :

Code:

res <- myfun.f(100)

Tu peux ensuite accéder à imc par

Code:

1
2
3
res$imc
## ou bien
res[["imc"]]

25/08/2011, 11h49
pitipoisson
Citation:
Envoyé par NemoParis

Code:

1 2 3 4 5 6 imc=rep(0,100) imc = sapply(1:100, function(i){ poids=rnorm(10,60,5) taille=rnorm(10,170,5) imc=poids/((taille/100)*(taille/100)) })
Quelques remarques au passage :
- Il est inutile de faire
  
  Code:
  
  imc=rep(0,100)
  
  dans ce cas-ci.
- Il est plus lisible d'utiliser " <- " pour les assignations (pas dans les arguments de fonctions par contre)
  
  Code:
  
  poids <- rnorm(10, 60, 5)
- Et enfin, il ne faut pas hésiter à utiliser des espaces pour aérer le code.
25/08/2011, 13h18
NemoParis

Merci pour vos conseils qui sont excellents (comme d'habitude!), je les retiendrai!
25/08/2011, 13h45
NemoParis

J'ai juste une question de curiosité :
comment faire si, par exemple, je veux calculer la variable essai=poids * taille à la place de la variable "fdr"? Car, du coup, je reprends les mêmes données qui ont été générées dans fonction imc...

En fait, il faut dans ce cas procéder autrement (à l'origine, je voulais utiliser des assign() dans la fonction du sapply()... ça ne pouvait pas marcher puisque tu ne récupérais que les valeurs de la dernière itération).

Le mieux à mon avis, c'est de créer les variables poids et taille dans des sapply séparés puis d'avoir recours aux possibilités de vectorisation :
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 myfun.f <- function(n) { poids <- sapply(1:n, function(x) rnorm(10, mean=60, sd=5)) taille <- sapply(1:n, function(x) rnorm(10, mean=170, sd=5)) ## IMC : imc <- poids / ((taille/100)^2) ## essai : essai <- poids * taille ## Retour des résultats : return(list(imc=imc, essai=essai)) }

26/08/2011, 13h08
NemoParis

:merci: beaucoup pitipoisson pour la réponse :ccool: