关于矩阵抽样的R代码

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

各位大神，我想要从一个100行10列的矩阵中按照按下面的代码中的P值从大到小的顺序提取前k行，这种R代码应该怎么写，是要构造一个循环吗？set.seed(123)
x=rnorm(1000)
y=matrix(x,100,10)
F=sqrt(sum(diag(t(y)%*%y)))
S=apply(abs(y),1,sum)
P=S^2/F^2

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：R代码

1. #简单写个函数
   
2. topK <- function (x,k) {
   
3.   if (k <= dim(x)[1]) x[which(rank(rowSums(x))>(dim(x)[1]-k)),]
   
4.   else
   
5.   print("Err: the specified k is out of matrix dimension")
   
6.   }
   
7. 
   
8. A<-matrix(runif(20,1,10),4) #示例4X5矩阵
   
9. 
   
10. topK(A,3) #提取矩阵A按行和排列前 k=3 子矩阵

复制代码

那个，function里的A改成x

llb_321 发表于 2020-12-23 20:07
那个，function里的A改成x

好的，谢谢您，我试一下。

llb_321 发表于 2020-12-23 20:07
那个，function里的A改成x

您好，我想给x加个绝对值，矩阵每行元素的绝对值求和，怎么就出错了
topK <- function (x,k) {
  if (k <= dim(x)[1]) x[which(rank(rowSums(abs(x))>(dim(x)[1]-k)),]
  else
    print("Err: the specified k is out of matrix dimension")
}

任潇潇 发表于 2020-12-23 20:28
您好，我想给x加个绝对值，矩阵每行元素的绝对值求和，怎么就出错了
topK

x[which(rank(rowSums(abs(x)))>(dim(x)[1]-k)),]
您改的代码里abs(x)后面少了个 ")"

另外如果按你后面提到的p值，可以用函数嵌套的方式

1. topK <- function (x, k) {
   
2.   pvalue <- function (x) {
   
3.     F = sqrt(sum(diag(t(x) %*% x)))
   
4.     S = apply(abs(x), 1, sum)
   
5.     P = S ^ 2 / F ^ 2
   
6.   }
   
7.   if (k <= dim(x)[1])
   
8.     x[which(rank(pvalue(x)) > (dim(x)[1] - k)), ]
   
9.   else
   
10.     print("Err: K is out of matrix dimension")
    
11. }
    
12. 
    
13. A <- matrix(rnorm(20, 0, 1), 4)
    
14. topK(A, 3)

复制代码

llb_321 发表于 2020-12-24 04:48
另外如果按你后面提到的p值，可以用函数嵌套的方式

谢谢您