[R] Ranking within factor subgroups
Adaikalavan Ramasamy
ramasamy at cancer.org.uk
Wed Feb 22 04:44:45 CET 2006
It might help to give a simple reproducible example in the future. For
example
df <- cbind.data.frame( date=rep( 1:5, each=100 ), A=rpois(500, 100),
B=rpois(500, 50), C=rpois(500, 30) )
might generate something like
date A B C
1 1 93 51 32
2 1 95 51 30
3 1 102 59 28
4 1 105 52 32
5 1 105 53 26
6 1 99 59 37
... . ... .. ..
495 5 100 57 19
496 5 96 47 44
497 5 111 56 35
498 5 105 49 23
499 5 105 61 30
500 5 92 53 32
Here is my proposed solution. Can you double check with your existing
functions to see if they are correct.
decile.fn <- function(x, nbreaks=10){
br <- quantile( x, seq(0, 1, len=nbreaks+1), na.rm=T )
br[1] <- -Inf
return( cut(x, br, labels=F) )
}
out <- apply( df[ ,c("A", "B", "C")], 2,
function(v) unlist( tapply( v, df$date, decile.fn ) ) )
rownames(out) <- rownames(df)
out <- cbind(df$date, out)
Regards, Adai
On Tue, 2006-02-21 at 21:44 -0500, maneesh deshpande wrote:
> Hi,
>
> I have a dataframe, x of the following form:
>
> Date Symbol A B C
> 20041201 ABC 10 12 15
> 20041201 DEF 9 5 4
> ...
> 20050101 ABC 5 3 1
> 20050101 GHM 12 4 2
> ....
>
> here A, B,C are properties of a set symbols recorded for a given date.
> I wante to decile the symbols For each date and property and
> create another set of columns "bucketA","bucketB", "bucketC" containing the
> decile rank
> for each symbol. The following non-vectorized code does what I want,
>
> bucket <- function(data,nBuckets) {
> q <- quantile(data,seq(0,1,len=nBuckets+1),na.rm=T)
> q[1] <- q[1] - 0.1 # need to do this to ensure there are no extra NAs
> cut(data,q,include.lowest=T,labels=F)
> }
>
> calcDeciles <- function(x,colNames) {
> nBuckets <- 10
> dates <- unique(x$Date)
> for ( date in dates) {
> iVec <- x$Date == date
> xx <- x[iVec,]
> for (colName in colNames) {
> data <- xx[,colName]
> bColName <- paste("bucket",colName,sep="")
> x[iVec,bColName] <- bucket(data,nBuckets)
> }
> }
> x
> }
>
> x <- calcDeciles(x,c("A","B","C"))
>
>
> I was wondering if it is possible to vectorize the above function to make it
> more efficient.
> I tried,
> rlist <- tapply(x$A,x$Date,bucket)
> but I am not sure how to assign the contents of "rlist" to their appropriate
> slots in the original
> dataframe.
>
> Thanks,
>
> Maneesh
>
> ______________________________________________
> R-help at stat.math.ethz.ch mailing list
> https://stat.ethz.ch/mailman/listinfo/r-help
> PLEASE do read the posting guide! http://www.R-project.org/posting-guide.html
>
More information about the R-help
mailing list