Question

我是R编程的初学者（使用它进行数据分析）

我有以下数据。（修剪版）

state   storeid sales
CA  1   40,000  
CA  2   44,000  
CA  3   38000   
MN  1   26000   
MN  2   25500

我需要一个能够返回顶级/低绩效商店的功能。

我写了以下函数。

storeinfo<-function(num="top") {

  df<-read.csv("store.csv")

  bestVal <- 1;
  if (!missing(num)) {
    if(is.numeric(num)){
      bestVal =  as.numeric(num);
    }
    if(num=="top"){
      bestVal <-1
    }
    if ( num=="poor"){
      bestVal<-0
    }
  }
  print(bestVal)
    data2<-subset(df[,c(1,2,3)])
    data2<-data2[order(as.numeric( data2$sales), data2$storeid,na.last=TRUE,decreasing=TRUE), ]
    idx<-tapply(1:NROW(data2),data2$state,"[",bestVal)
    idx1<-tapply(1:NROW(data2),data2$state,"[",1)

    return (data.frame(data2[idx1,1],data2[idx,2:3]))

}

当我执行上述功能时，我看到以下

> head(storeinfo(1))
[1] 1
  data2.idx1..1. storeid  sales
2             CA       2 44,000
4             MN       1  26000

a）如何压制第一列2,4等？（索引） b）如何找到销量低的商店？ c）如何为返回的数据框设置不同的列名。

Answer 1

这样做

min_max_sales = c(which.min(df$sales), which.max(df$sales)) # return row numbers
df[min_max_sales,]

更新。

如果您需要对数据框进行排序并获得具有第n个最佳销售额的商店，那么使用plyr包

是更好的方法

plyr::arrange(df, sales)[n,]

R函数返回数据帧（额外列）

1 个答案: