2013-05-14 23 views
5

Se ho un data.frame (sum_clus) con 600 colonne (variabili) e 10 righe che non hanno NA e sono tutti valori numerici, come posso creare 5 nuove variabili che mi danno i nomi delle colonne delle 5 variabili principali in quella fila?Come ottenere il nome della colonna della variabile con i primi 10 valori più alti?

Ad es.

max <- apply(sum_clus ,1, max)  
for(ii in 1:10) sum_clus$max[ii] <- colnames(sum_clus)[which(sum_clus[ii , ] 
== sum_clus[ii, sum_clus[ii,] == max[ii]])] 

Questo codice di cui sopra mi ha aiutato a creare una variabile sum_clus $ max che mi dà il nome della colonna della variabile massimo in ogni riga. Allo stesso modo, come posso ottenere 5 variabili di questo tipo che mi danno i nomi delle colonne delle 5 variabili principali? sum_clus $ max, sum_clus $ second_but_max, e così via ..

Grazie in anticipo!

+1

Potrebbe fornire un campione? Forse 'sum_clus [1: 10,1: 10]' o qualcosa del genere ...? – Frank

risposta

3

Ecco una soluzione simile, utilizzando (i) un ciclo anziché apply; e (ii) rank anziché order.

set.seed(1) 
n_i = 10 
n_ii = 600 
n_top = 5 
df <- data.frame(matrix(runif(n_ii*n_i), ncol = n_ii)) 

out <- matrix("",n_top,n_i) 
for (i in 1:n_i){ 
    colranks <- rank(df[i,]) 
    out[,i] <- names(sort(colranks)[n_ii:(n_ii-(n_top-1))]) 
} 
#  [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] 
# [1,] "X369" "X321" "X348" "X415" "X169" "X258" "X55" "X182" "X99" "X78" 
# [2,] "X42" "X295" "X563" "X173" "X377" "X31" "X246" "X353" "X259" "X384" 
# [3,] "X98" "X440" "X371" "X207" "X429" "X292" "X433" "X437" "X123" "X558" 
# [4,] "X13" "X193" "X396" "X78" "X543" "X228" "X211" "X2" "X583" "X508" 
# [5,] "X35" "X364" "X249" "X33" "X388" "X405" "X458" "X252" "X569" "X456" 

L'analogo one-liner con apply è

apply(df,1,function(x)names(sort(rank(x))))[600:596,] 
+0

dov'è il tuo loop? –

+0

@RicardoSaporta Ho sovrascritto la mia soluzione con una risposta completa ora. – Frank

+2

Mi è piaciuto di più senza il ciclo;) –

5

Una possibilità è quella di utilizzare order() e quindi utilizzare tale per sottoinsieme i nomi delle colonne, ad esempio:

set.seed(1) 
df <- data.frame(matrix(runif(600*10), ncol = 600)) 

foo <- function(x, names) { 
    ord <- order(x, decreasing = TRUE)[1:5] 
    names[ord] 
} 

nams <- colnames(df) 
apply(df, 1, foo, names = nams) 

Produrre

> apply(df, 1, foo, names = nams) 
    [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10] 
[1,] "X369" "X321" "X348" "X415" "X169" "X258" "X55" "X182" "X99" "X78" 
[2,] "X42" "X295" "X563" "X173" "X377" "X31" "X246" "X353" "X259" "X384" 
[3,] "X98" "X440" "X371" "X207" "X429" "X292" "X433" "X437" "X123" "X558" 
[4,] "X13" "X193" "X396" "X78" "X543" "X228" "X211" "X2" "X583" "X508" 
[5,] "X35" "X364" "X249" "X33" "X388" "X405" "X458" "X252" "X569" "X456" 

Controllare questo funziona:

> names(sort(unlist(df[1,, drop = TRUE]), decreasing = TRUE)[1:5]) 
[1] "X369" "X42" "X98" "X13" "X35" 
> names(sort(unlist(df[2,, drop = TRUE]), decreasing = TRUE)[1:5]) 
[1] "X321" "X295" "X440" "X193" "X364" 

sembra ok .

Problemi correlati